update config to support colab/local execution as starting point

config.yaml

@@ -1,8 +1,8 @@
 wandb_mode: "disabled"
 vartype: "uvtp122"
-train_data_path: "gravity_wave_flux/uvtp122"
-valid_data_path: "gravity_wave_flux/uvtp122/test"
-singular_sharded_checkpoint: "prithvi_wxc/v0.8.50.rollout_step3.1.pth"
+train_data_path: "."
+valid_data_path: "."
+singular_sharded_checkpoint: "magnet-flux-uvtp122-epoch-99-loss-0.1022.pt"
 file_glob_pattern: "wxc_input_u_v_t_p_output_theta_uw_vw_era5_*.nc"
 
 lr: 0.0001
@@ -20,4 +20,4 @@ patch_size_px:
 ### Training Params
 max_epochs: 100
 batch_size: 12
-num_data_workers: 8
+num_data_workers: 8