File size: 385 Bytes
bd3493c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | python3 ../aardvark/train_module.py \
--output_dir ENCODER_PATH \
--master_port 12348 \
--decoder vit_assimilation \
--loss lw_rmse \
--diff 0 \
--in_channels 277 \
--out_channels 24 \
--int_channels 24 \
--mode assimilation \
--lr 5e-4 \
--batch_size 6 \
--start_ind 0 \
--end_ind 24 \
--epoch 100 \
--weight_per_variable 1
|