train_shapenet.yamlΒΆ

recipes/aero_cfd/configs/train_shapenet.yaml

 1# @package _global_
 2
 3# Define key values here that are used multiple times in the config files.
 4dataset_root: <path to your shapenet dataset root>
 5dataset_kind: noether.data.datasets.cfd.ShapeNetCarDataset
 6excluded_properties: 
 7  - surface_friction
 8  - volume_pressure
 9  - volume_vorticity
10defaults:
11  - data_specs: shapenet_car
12  - dataset_normalizers: shapenet_dataset_normalizers
13  - model: ??? # models are undefined and will be defined per experiment 
14  - trainer: shapenet_trainer
15  - datasets: shapenet_dataset
16  - tracker: ?? # trackers are undefined and will be defined depending on either development, training or evaluation
17  - callbacks: training_callbacks_shapenet
18  - pipeline: shapenet_pipeline
19  - optimizer: adamw
20  - slurm: slurm_config 
21  - _self_
22
23stage_name: train
24store_code_in_output: false
25output_path: ./outputs
26
27
28# to continue training from a previous run checkpoint (which has not eneded yet), specify the run_id and stage_name to resume from. 
29#The model will be initialized from the lastest checkpoint of the specified stage if no checkpoint is specified.
30#resume_run_id: 2026-02-19_f7dy0
31#resume_stage_name: train
32#` stage_name: continue_training # optional, only needed if you want to use a different stage name for the continued training run. If not specified, the stage name will be the same as the stage name of the previous run (train in this case).
33#resume_checkpoint: E3 #If a full checkpoint path is E3_U5523_S5523. Either E*, U*, S* need to be provided
34
35# example of how to use the PreviousRunInitializer to initialize the model from a previous run.
36
37#model:
38#  initializers:
39#    - kind: noether.core.initializers.PreviousRunInitializer
40#      run_id: 2026-02-19_f7dy0
41#      model_name: transformer
42#      stage_name: train # only when a stage name is used for the previous run
43#      checkpoint_tag: E4_U3156_S3156 # can be `latest`, `last`, `best`, or a specific epoch like E3_U100_S1000
44#      model_info: ema=0.9999 # only needed when a checkoint has additional model info