# 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"