shapenet_trainer.yamlΒΆ

recipes/aero_cfd/configs/trainer/shapenet_trainer.yaml

 1#BaseTrainerConfig
 2kind: trainers.AerodynamicsCFDTrainer
 3precision: bfloat16
 4max_epochs: 500
 5effective_batch_size: 1
 6log_every_n_epochs: 1
 7callbacks:  ${callbacks} 
 8forward_properties: ${model.forward_properties}
 9target_properties:
10  - surface_pressure_target
11  - volume_velocity_target
12# AerodynamicsCFDTrainerConfig
13surface_weight: 1.0
14volume_weight: 1.0 
15surface_pressure_weight: 1.0 
16volume_velocity_weight: 1.0 
17use_physics_features: false # whether to use the physics features (e.g., surface normals, volume normals, SDF) as input to the model