File size: 443 Bytes
d2693e0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
export CUDA_VISIBLE_DEVICES=5,6
python -u ./smiles/rectify_train.py \
--checkpoint ./smiles/ckpt/SMILESReDi_base.ckpt \
--gamma 2.0 \
--epochs 10 \
--validity_start_epoch 10 \
--validity_weight_min 0 \
--validity_weight_max 0 \
--validity_schedule linear \
--time_dependent_validity False \
--learning_rate 1e-4 \
--weight_decay 1e-5 \
--model_dim 768 \
--n_heads 8 \
--n_layers 8 \
--checkpoint_dir ./smiles/ckpt
|