update

data_ljkj/results/tec_256/afno_backbone_ljkj/d6p4_tt/hyperparams.yaml

@@ -2,7 +2,7 @@ log_to_wandb: 'False'
 lr: '0.0005'
 batch_size: '8'
 patch_size: '4'
-depth: '6'
+depth: '12'
 img_size: '[192, 288]'
 max_epochs: '1500'
 scheduler: CosineAnnealingLR

data_ljkj_6h_720/AFNO.yaml

@@ -66,8 +66,8 @@ afno_backbone_ljkj: &LJKJ
   lr: 5E-4
   batch_size: 16
   patch_size: 16
-  depth : 6 # default 12
-  img_size: [192, 288]
+  depth : 12 # default 12
+  img_size: [720, 1440]
   max_epochs: 1500
   scheduler: 'CosineAnnealingLR'
   in_channels: [0, 1 ,2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29]
@@ -86,4 +86,3 @@ afno_backbone_ljkj: &LJKJ
 afno_backbone_ustc:
   <<: *LJKJ
   batch_size: 32
-  depth: 12
\ No newline at end of file