#!/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='tr2d2-finetune-tfr'
PYTHON_EXECUTABLE=$ENV_PATH/bin/python

# ===================================================================

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

# ===================================================================

$PYTHON_EXECUTABLE $SCRIPT_LOC/finetune.py \
    --base_path $HOME_LOC \
    --device "cuda:0" \
    --noise_removal \
    --wdce_num_replicates 16 \
    --buffer_size 20 \
    --seq_length 200 \
    --num_children 50 \
    --total_num_steps 128 \
    --num_iter 10 \
    --resample_every_n_step 10 \
    --num_epochs 1000 \
    --exploration 0.1 \
    --save_every_n_epoch 50 \
    --reset_every_n_step 1 \
    --alpha 0.1 \
    --grad_clip > "${LOG_LOC}/${DATE}_${SPECIAL_PREFIX}.log" 2>&1

conda deactivate