File size: 1,083 Bytes
7efee70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
#!/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-150D'
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_150D.yaml" \
--save_dir "$SCRIPT_LOC/results/$DATE-clonidine-ce-150D" \
--root_dir "$SCRIPT_LOC" \
--wandb \
--device "cuda:0" \
--run_name "clonidine-ce-150D" \
--date "$DATE-clonidine-ce-150D" \
--num_rollouts 100 \
--num_steps 100 \
--use_delta_to_target \
--objective "ce" \
--sigma 0.1 \
--num_samples 64 \
--self_normalize \
--kT 0.1 \
--unseen \
--num_particles 16 \
--batch_size 64 \
--friction 2.0 \
--vel_conditioned >> ${LOG_LOC}/${DATE}_${SPECIAL_PREFIX}_train.log 2>&1
conda deactivate |