| export CUDA_VISIBLE_DEVICES=3,4,5,6 | |
| python -u ./smiles/train.py \ | |
| --validity_weight 200 \ | |
| --gamma 2.0 \ | |
| --epochs 5 \ | |
| --learning_rate 1e-4 \ | |
| --weight_decay 1e-5 \ | |
| --model_dim 768 \ | |
| --n_heads 8 \ | |
| --n_layers 8 \ | |
| --checkpoint_dir ./smiles/ckpt | |
| export CUDA_VISIBLE_DEVICES=3,4,5,6 | |
| python -u ./smiles/train.py \ | |
| --validity_weight 200 \ | |
| --gamma 2.0 \ | |
| --epochs 5 \ | |
| --learning_rate 1e-4 \ | |
| --weight_decay 1e-5 \ | |
| --model_dim 768 \ | |
| --n_heads 8 \ | |
| --n_layers 8 \ | |
| --checkpoint_dir ./smiles/ckpt | |