| # GeoForce CNN Surrogate — Training Hyperparameters | |
| model: | |
| architecture: "ReservoirCNN" | |
| params: 56000 | |
| grid_size: 32 | |
| in_channels: 5 # T, P, permeability, porosity, depth | |
| out_channels: 2 # T_predicted, P_predicted | |
| n_timesteps: 5 | |
| training: | |
| epochs: 500 | |
| learning_rate: 1.0e-3 | |
| batch_size: 32 | |
| optimizer: "Adam" | |
| scheduler: "ReduceLROnPlateau" | |
| physics_weight: 0.1 | |
| early_stopping_patience: 50 | |
| data: | |
| source: "tough2_simulations" | |
| train_split: 0.8 | |
| val_split: 0.1 | |
| test_split: 0.1 | |
| compute: | |
| device: "cpu" | |
| estimated_time: "30 minutes" | |
| platform: "vps" | |