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 \