-
Notifications
You must be signed in to change notification settings - Fork 14
/
Copy pathrun.sh
11 lines (11 loc) · 1.52 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
if [ ! -d "./logs" ]; then
mkdir ./logs
fi
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model GRU -num_head 8 --tau 10 --lr 0.01 --market csi300 --data_dir cn_data --rank_label False > logs/gru300.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model LSTM -num_head 8 --tau 10 --lr 0.01 --market csi300 --data_dir cn_data --rank_label False > logs/lstm300.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model ALSTM -num_head 8 --tau 10 --lr 0.01 --market csi300 --data_dir cn_data --rank_label False > logs/alstm300.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model Transformer -num_head 8 --tau 10 --lr 0.01 --market csi300 --data_dir cn_data --rank_label False > logs/tfm300.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model GRU -num_head 8 --tau 10 --lr 0.01 --market csi500 --data_dir cn_data --rank_label False > logs/gru500.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model LSTM -num_head 8 --tau 10 --lr 0.01 --market csi500 --data_dir cn_data --rank_label False > logs/lstm500.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model ALSTM -num_head 8 --tau 10 --lr 0.01 --market csi500 --data_dir cn_data --rank_label False > logs/alstm500.log 2>&1
CUDA_VISIBLE_DEVICES=0 python -u main.py workflow --forecast_model Transformer -num_head 8 --tau 10 --lr 0.001 --market csi500 --data_dir cn_data --rank_label False > logs/tfm500.log 2>&1