conf/config.yaml

defaults:
  - modulus_default
  - arch:
      - fourier_net
  - optimizer: adam
  - scheduler: tf_exponential_lr
  - loss: sum
  - _self_

jit: false
save_filetypes: "vtk"

custom:
  parameterized: true
  turbulent: true

################################################################################
# ARCHITECTURE
################################################################################
arch:
  fourier_net:
    arch_type: "fourier"
    _target_: "modulus.sym.models.fourier_net.FourierNetArch"

    # Input/Output dims
    input_key_dims: [3, 5]        # (x,y,z) + (bld_x,bld_y,bld_width,bld_depth,bld_height)
    output_key_dims: [3, 1]       # (u,v,w) + p

    layer_size: 256
    num_layers: 4
    activation_fn: silu
    weight_norm: true
    adaptive_activations: false

################################################################################
# OPTIMIZER
################################################################################
optimizer:
  lr: 0.001
  betas: [0.9, 0.999]
  eps: 1.0e-8
  weight_decay: 1.0e-5

################################################################################
# SCHEDULER
################################################################################
scheduler:
  decay_rate: 0.98
  decay_steps: 1000

################################################################################
# LOSS
################################################################################
loss:
  weights:
    inlet: 20.0              # Strongly enforce inlet conditions
    outlet: 5.0              # Increased to better maintain outflow
    no_slip_building: 5.0    # Keep building interaction
    top: 0.5                 # Further reduced wall influence
    ground: 0.5              # Further reduced wall influence
    sides_y0: 0.5            # Further reduced wall influence
    sides_y5: 0.5            # Further reduced wall influence
    interior: 10.0           # Strongly enforce momentum equations in interior

################################################################################
# TRAINING
################################################################################
training:
  max_steps: 40000
  rec_results_freq: 100
  rec_constraint_freq: 5000
  save_network_freq: 5000

################################################################################
# BATCH SIZES
################################################################################
batch_size:
  inlet: 1024
  outlet: 1024
  no_slip_building: 2048
  slip: 1024
  interior: 4096