update

data_ljkj_6h_720/AFNO.yaml

@@ -64,8 +64,8 @@ afno_backbone_ljkj: &LJKJ
   <<: *FULL_FIELD
   log_to_wandb: !!bool False
   lr: 5E-4
-  batch_size: 8
-  patch_size: 4
+  batch_size: 16
+  patch_size: 16
   depth : 6 # default 12
   img_size: [192, 288]
   max_epochs: 1500

data_ljkj_6h_720/collect_data.py

@@ -4,7 +4,7 @@ from time import time
 import os 
 os.getcwd()
 start_date = datetime(2010, 1, 11)
-end_date = datetime(2010, 1, 21)
+end_date = datetime(2010, 12, 31)
 
 kind = 'Density'
 value = []