source_copy/recover_schedules.bash

python recover_schedules.py\
    --testing True\
    --seed 97\
    --init_mode random\
    --n_optimization_steps 2000\
    --epsilon 0.01\
    --nb_sinkhorn_iters 40\
    --decay_ls ""\
    --ls_partitions_ratios 0.66\
    --init_ls 2\
    --max_ls 10\
    --min_ls 0.1\
    --ls_warmup_iters_ratio 0.2\
    --ls_decay_iters_ratio 0.90\
    --decay_lr "True"\
    --lr_partitions_ratios 0.66\
    --init_lr 1e-2\
    --max_lr 1e-1\
    --min_lr 1e-4\
    --lr_warmup_iters_ratio 0.1\
    --lr_decay_iters_ratio 0.90\
    --checkpoint_interval 1\
    --data_dir "./demos/ftd_processed"\
    --model_dir "./demos/train_artifacts"\
    --output_dir "./demos/rs_artifacts"