File size: 859 Bytes
ab687e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
#!/bin/bash
#SBATCH -J finetune_satvision_lc9
#SBATCH -t 3-00:00:00
#SBATCH -G 4
#SBATCH -N 1
export PYTHONPATH=$PWD:../../../:../../../pytorch-caney
export NGPUS=8
torchrun --nproc_per_node $NGPUS \
../../../pytorch-caney/pytorch_caney/pipelines/finetuning/finetune.py \
--cfg finetune_satvision_base_landcover5class_192_window12_100ep.yaml \
--pretrained /explore/nobackup/people/cssprad1/projects/satnet/code/development/masked_image_modeling/development/models/simmim_satnet_pretrain_pretrain/simmim_pretrain__satnet_swinv2_base__img192_window12__800ep_v3_no_norm/ckpt_epoch_800.pth \
--dataset MODISLC9 \
--data-paths /explore/nobackup/projects/ilab/data/satvision/finetuning/h18v04/labels_5classes_224 \
--batch-size 4 \
--output /explore/nobackup/people/cssprad1/projects/satnet/code/development/cleanup/finetune/models \
--enable-amp |