| data: | |
| seq_len: 12 | |
| pred_horz: 12 | |
| batch_size: 128 | |
| shuffle: False | |
| dataSplit: chronologicalSplit # Chronological or Stratified Refer to ~/Projects/PhD/Ionosphere/Python/Data Scripts/Preprocessing/Preprocess_Data_v3.ipynb | |
| selectedOMNI: [1,4,5,6,9,10,11,12,13,14,16,17,19,27,28,29,30] # Check data.py | |
| # ['F10.7', 'Dst', 'ap', 'AE', 'pc', 'SW Temp', 'SW Density', 'SW Speed', 'SW Long Angle', 'SW Lat Angle', 'Flow Pressure', 'E Field', 'Scalar B', 'BZ (GSM)', 'Year', 'Day of Year', 'Hour'] | |
| # [ 1, 4, 5, 6, 9, 10, 11, 12, 13, 14, 16, 17, 19, 27, 28, 29, 30] | |
| model: | |
| type: "SimVP" | |
| input_shape: [12, 18, 72, 72] # (Time, Channels, Height, Width) | |
| hid_S: 16 | |
| hid_T: 256 | |
| N_S: 4 | |
| N_T: 4 | |
| model_type: "gSTA" | |
| mlp_ratio: 8.0 | |
| drop: 0.0 | |
| drop_path: 0.0 | |
| spatio_kernel_enc: 3 | |
| spatio_kernel_dec: 3 | |
| conv_in_channels: 18 | |
| conv_out_channels: 1 | |
| conv_kernel_size: [1, 1, 1] | |
| conv_stride: [1, 1, 1] | |
| train: | |
| model: "SimVP" # Model to train | |
| sessionName: "SimVP_chronologicalSplit" # Name of the session | |
| lr: 8e-5 # Learning rate for Optimizer (LRrangetest minima at [LR=0.0001] Global average loss = 5.77232401785838e-06) | |
| lrEnd: 5e-6 # Learning rate end for ExponentialLR | |
| batchSize: 128 # Batch size for training | |
| numEpochs: 50 # Number of epochs to train | |
| resumeEpoch: 0 # Resume training from epoch | |
| earlyStop: 10 # Early stopping patience | |
| saveResults: True # Save results | |
| inputNames: [] # Names of the input features | |
| criterion: "MSELoss" # Loss functions: MSELoss. | |
| optimizer: "Adam" # Optimizer: Adam. | |
| scheduler: "ReduceLROnPlateau" # Scheduler: ExponentialLR, ReduceLROnPlateau | |
| mixedPrecision: False # Mixed precision training | |
| test: | |
| model: "SimVP" | |
| sessionName: "SimVP_chronologicalSplit" # Name of the session | |
| batchSize: 128 | |
| testDates: [] # Dates to test will be set in the script | |
| maxTEC: "210" # Set inside the main script | |
| minTEC: "0" # Set inside the main script | |
| lat: "72" # Set inside the main script | |
| lon: "72" # Set inside the main script | |
| numOMNI: "18" # Set inside the main script | |
| inputNames: [] # Set inside the main script | |
| saveResults: True |