Upload training config

config.yml

@@ -0,0 +1,65 @@
+data_dir: /scratch/project_2015260/data/fold/fold_generated_meshes_seed_1397.h5
+output_dir: /scratch/project_2015260/outputs/dynamics_gps/2026-05-04/16-54-43_6616925
+logging_dir: logs/
+gradient_accumulation_steps: 2
+mixed_precision: bf16
+report_to: wandb
+seed: 259
+use_ema: false
+gradient_checkpointing: false
+scale_lr: false
+allow_tf32: true
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_weight_decay: 0.01
+adam_epsilon: 1.0e-08
+max_grad_norm: 1.0
+per_gpu_batch_size: 256
+learning_rate: 0.0003
+num_workers: 8
+compile_model: false
+lr_scheduler: cosine
+lr_warmup_steps: 2000
+max_train_steps: 900000
+resume_from_checkpoint: null
+pretrained_model_name_or_path: null
+checkpointing_steps: 2000
+sampling_steps: 1000
+checkpoints_total_limit: 10
+exp_name: GPSDynamics_d256_H8_F5_L8_lr0.0003_bs256_fold_04052026_1656
+num_train_epochs: 999999
+config: configs/train_dynamics_gps.yaml
+do_classifier_free_guidance: false
+val_inference_steps: 50
+model_cfg:
+  type: GPSDynamicsModel
+  num_prev_frames: 3
+  num_next_frames: 5
+  action_dim: 3
+  hidden_dim: 256
+  num_layers: 8
+  num_heads: 8
+  edge_feature_dim: 4
+  max_num_nodes: 160
+  dropout: 0.1
+  num_embeds_ada_norm: 1000
+diffusion_cfg:
+  type: DDPM_DynamicsGPS
+  num_train_timesteps: 1000
+  beta_schedule: scaled_linear
+  snr_gamma: 5.0
+  history_noise_std: 0.0001
+wandb_cfg:
+  project_name: cloth_sim_mujoco
+  entity: null
+  tags:
+  - gps
+  - dynamics
+dataset_cfg:
+  type: ClothDynamicsVariableDataset
+  data_dir: /scratch/project_2015260/data/fold/fold_generated_meshes_seed_1397.h5
+  num_prev_frames: 3
+  num_next_frames: 5
+  max_num_nodes: 160
+  max_num_edges: 832
+  val_trajectory_ratio: 0.1