update

results/tec_256/afno_backbone_tec_256/d6p_test1/hyperparams.yaml

+log_to_wandb: 'False'
+lr: '0.0005'
+batch_size: '8'
+patch_size: '2'
+depth: '6'
+img_size: '[128, 256]'
+max_epochs: '1500'
+scheduler: CosineAnnealingLR
+in_channels: '[ 0  1  2  3  4  5  6  7  8  9 10 11 12]'
+out_channels: '[ 0  1  2  3  4  5  6  7  8  9 10 11 12]'
+prediction_length: '100'
+orography: 'False'
+orography_path: None
+exp_dir: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/results/tec_256
+train_data_path: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/Data_TEC_256x512/train
+valid_data_path: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/Data_TEC_256x512/test
+inf_data_path: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/Data_TEC_256x512/out_of_sample
+time_means_path: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/Data_TEC_256x512/time_means.npy
+global_means_path: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/Data_TEC_256x512/global_means.npy
+global_stds_path: /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/Data_TEC_256x512/global_stds.npy
+loss: l2
+num_data_workers: '4'
+dt: '1'
+n_history: '0'
+prediction_type: iterative
+n_initial_conditions: '5'
+ics_type: default
+save_raw_forecasts: 'True'
+save_channel: 'False'
+masked_acc: 'False'
+maskpath: None
+perturb: 'False'
+add_grid: 'False'
+N_grid_channels: '0'
+gridtype: sinusoidal
+roll: 'False'
+num_blocks: '8'
+nettype: afno
+width: '56'
+modes: '32'
+target: default
+normalization: zscore
+log_to_screen: 'True'
+save_checkpoint: 'True'
+enable_nhwc: 'False'
+optimizer_type: FusedAdam
+crop_size_x: None
+crop_size_y: None
+two_step_training: 'False'
+plot_animations: 'False'
+add_noise: 'False'
+noise_std: '0'
+epsilon_factor: '0'
+world_size: '1'
+experiment_dir: 
+  /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/results/tec_256/afno_backbone_tec_256/d6p_test1
+checkpoint_path: 
+  /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/results/tec_256/afno_backbone_tec_256/d6p_test1/training_checkpoints/ckpt.tar
+best_checkpoint_path: 
+  /home/cxt/work/fourcastnet_TEC/FourCastNetTEC/results/tec_256/afno_backbone_tec_256/d6p_test1/training_checkpoints/best_ckpt.tar
+resuming: 'False'
+local_rank: '0'
+enable_amp: 'True'
+name: afno_backbone_tec_256_d6p_test1
+group: era5_precipafno_backbone_tec_256
+project: ERA5_precip
+entity: flowgan
+N_in_channels: '13'
+N_out_channels: '13'