| CUDA_VISIBLE_DEVICES=4,5,6,7 torchrun --nproc_per_node=4 \ | |
| train_aurora_lora.py \ | |
| --base_dir /hub_data3/minseo/ArchesWeather-SR/data/era5_240/full \ | |
| --train_years 2016-2020 --val_years 2021 \ | |
| --steps 1 --total_steps 10000 --warmup_steps 1000 \ | |
| --lr 5e-5 --weight_decay 0 \ | |
| --precision bf16 --batch_size 1 \ | |
| --act_ckpt --disable_droppath \ | |
| --out_dir /hub_data3/minseo/aurora/lora_ckpt_wloss_hres/identity \ | |
| --tb_dir /hub_data3/minseo/aurora/tb_lora_wloss_hres/identity \ | |
| --deg_old 1.5 --deg_new 1.5 --win_orig 2,4,8 --win_min_hw 1 --patch_new 1 \ | |
| --finetune_mode lora --scheduler cosine --min_lr_scale 0.05 \ | |
| --surf_stats_json /hub_data3/minseo/aurora/surf_stats_1p5deg.json \ | |
| --surf_stats_days 1000 \ | |
| --tasks identity \ | |
| --adapter_mode lora \ | |
| --init_ckpt /hub_data3/minseo/aurora/lora_ckpt_wloss_hres/full_small_final.pt \ | |
| --init_ckpt_p_old 1 \ | |
| --enable_ckpt_adapt --enable_window_retarget |