File size: 823 Bytes
dc3477c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | TRAIN_START_DATE='1998-01-01'
TRAIN_END_DATE='2015-12-31'
VAL_START_DATE="2016-01-03"
VAL_END_DATE="2017-12-31"
OUTPUT_DIR="output/WeatherPEFT"
OMP_NUM_THREADS=1 python -m torch.distributed.run --nproc_per_node=8 \
--master_port 12376 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" \
run_postprocessing.py \
--mode ours \
--log_dir ${OUTPUT_DIR} \
--train_start_date ${TRAIN_START_DATE} \
--train_end_date ${TRAIN_END_DATE} \
--val_start_date ${VAL_START_DATE} \
--val_end_date ${VAL_END_DATE} \
--output_dir ${OUTPUT_DIR} \
--batch_size 1 \
--val_batch_size 4 \
--save_ckpt_freq 5 \
--opt adamw \
--lr 1e-3 \
--opt_betas 0.9 0.999 \
--weight_decay 0.05 \
--warmup_epochs 3 \
--epochs 10 \
--dist_eval \
--clip_grad 5 \
# --eval \
|