Spaces:
Configuration error
Configuration error
File size: 2,528 Bytes
81efd79 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
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
|