File size: 767 Bytes
5e90249 4e90f71 5e90249 |
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 |
#!/bin/bash
HOME_LOC=/path/to/your/home
ENV_PATH=/path/to/your/env
SCRIPT_LOC=$HOME_LOC/TR2-D2/tr2d2-pep
LOG_LOC=$HOME_LOC/TR2-D2/tr2d2-pep/logs
DATE=$(date +%m_%d)
SPECIAL_PREFIX='tfr-peptune-baseline'
PYTHON_EXECUTABLE=$ENV_PATH/bin/python
# ===================================================================
source "$(conda info --base)/etc/profile.d/conda.sh"
conda activate $ENV_PATH
$PYTHON_EXECUTABLE $SCRIPT_LOC/generate_mcts.py \
--base_path $HOME_LOC \
--device "cuda:0" \
--noise_removal \
--run_name "tfr-peptune-baseline" \
--num_children 50 \
--num_iter 100 \
--buffer_size 100 \
--seq_length 200 \
--total_num_steps 128 \
--exploration 0.1 > "${LOG_LOC}/${DATE}_${SPECIAL_PREFIX}.log" 2>&1
conda deactivate |