update

data_ljkj_6h_90_un/AFNO.yaml

@@ -70,10 +70,10 @@ afno_backbone_ljkj: &LJKJ
   img_size: [192, 288]
   max_epochs: 1500
   scheduler: 'CosineAnnealingLR'
-  in_channels_range: [ 0, 162, 1 ]
-  out_channels_range: [ 0, 162, 1 ]
-#  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]
-#  out_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]
+#  in_channels_range: [ 0, 162, 1 ]
+#  out_channels_range: [ 0, 162, 1 ]
+  in_channels: [0, 1 ,2, 3, 4, 5]
+  out_channels: [0, 1 ,2, 3, 4, 5]
   prediction_length: 100
   orography: !!bool False
   orography_path: None
@@ -87,6 +87,6 @@ afno_backbone_ljkj: &LJKJ
 
 afno_backbone_ustc:
   <<: *LJKJ
-  batch_size: 8
-  depth: 12
+  batch_size: 64
+  depth: 6
   patch_size: 2
\ No newline at end of file