#!/bin/bash

HOME_LOC=/path/to/your/home/EntangledSBM
ENV_LOC=/path/to/your/envs/entangled-cell
SCRIPT_LOC=$HOME_LOC/entangled-cell
LOG_LOC=$HOME_LOC/entangled-cell/logs
DATE=$(date +%m_%d)
SPECIAL_PREFIX='clonidine-ce-100D'
PYTHON_EXECUTABLE=$ENV_LOC/bin/python

# ===================================================================
source "$(conda info --base)/etc/profile.d/conda.sh"
conda activate $ENV_LOC

$PYTHON_EXECUTABLE $SCRIPT_LOC/train_cell.py \
    --config_path "$SCRIPT_LOC/configs/clonidine_100D.yaml" \
    --save_dir "$SCRIPT_LOC/results/$DATE-clonidine-ce-100D" \
    --root_dir "$SCRIPT_LOC" \
    --wandb \
    --device "cuda:0" \
    --run_name "clonidine-ce-100D" \
    --date "$DATE-clonidine-ce-100D" \
    --num_rollouts 100 \
    --num_steps 100 \
    --use_delta_to_target \
    --objective "ce" \
    --sigma 0.1 \
    --num_samples 64 \
    --self_normalize \
    --kT 0.05 \
    --unseen \
    --num_particles 16 \
    --batch_size 64 \
    --friction 2.0 \
    --vel_conditioned >> ${LOG_LOC}/${DATE}_${SPECIAL_PREFIX}_train.log 2>&1

conda deactivate