-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_formation_energy_7-9.sh
5 lines (5 loc) · 1.16 KB
/
train_formation_energy_7-9.sh
1
2
3
4
5
#!/bin/bash
nohup python -u train.py --gpu 0 --seed 0 --data_path ./data/material_project --train_path ./data/material_project --dataset_name formation_energy --dataset_type regression --run_fold 7 --metric mae --save_dir ./ckpt/ensemble_formation_energy --epochs 100 --init_lr 1e-4 --max_lr 3e-4 --final_lr 1e-4 --no_features_scaling --show_individual_scores > ./log/fold_7_formation_energy.log 2>&1 &
nohup python -u train.py --gpu 1 --seed 0 --data_path ./data/material_project --train_path ./data/material_project --dataset_name formation_energy --dataset_type regression --run_fold 8 --metric mae --save_dir ./ckpt/ensemble_formation_energy --epochs 100 --init_lr 1e-4 --max_lr 3e-4 --final_lr 1e-4 --no_features_scaling --show_individual_scores > ./log/fold_8_formation_energy.log 2>&1 &
nohup python -u train.py --gpu 2 --seed 0 --data_path ./data/material_project --train_path ./data/material_project --dataset_name formation_energy --dataset_type regression --run_fold 9 --metric mae --save_dir ./ckpt/ensemble_formation_energy --epochs 100 --init_lr 1e-4 --max_lr 3e-4 --final_lr 1e-4 --no_features_scaling --show_individual_scores > ./log/fold_9_formation_energy.log 2>&1 &
wait