update

data_ljkj/AFNO.yaml

@@ -86,4 +86,5 @@ afno_backbone_ljkj: &LJKJ
 afno_backbone_ustc:
   <<: *LJKJ
   batch_size: 32
-  depth: 12
\ No newline at end of file
+  depth: 6
+  patch_size: 2
\ No newline at end of file

data_ljkj/submit_batch.sh

@@ -6,7 +6,7 @@
 
 config_file=./AFNO.yaml
 config='afno_backbone_ustc'
-run_num='d12p4'
+run_num='d6p2'
 
 export HDF5_USE_FILE_LOCKING=FALSE
 export NCCL_NET_GDR_LEVEL=PHB