config.yaml

# ignore_header_test
# ignore_header_test
# ruff: noqa: E402

# © Copyright 2023 HP Development Company, L.P.
# SPDX-FileCopyrightText: Copyright (c) 2023 - 2026 NVIDIA CORPORATION & AFFILIATES.
# SPDX-FileCopyrightText: All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


general:
  mode: "rollout"   # Train model, one step evaluation or rollout evaluation, options: ['train', 'eval_rollout', 'rollout']
  eval_split: "inference"  # Stored output dataset name, options: ['train', 'valid', 'test']
  device: "cuda:0"
  message_passing_devices: "['cuda:0']"
  fp16: False     # performance configs

train_options:
    batch_size: 1
    num_steps: 2e7
    eval_steps: 1
    prefetch_buffer_size: 100
    input_seq_len: 5 # calculate the last 5 velocities. [options: 5, 10]
    pred_len: 1 # [options: 5]
    loss: "me"
    loss_decay_factor: 0.6  #range (0, 1]
    l_plane: 30
    l_me: 3

data_options:
  data_path: "./data/2024-3-absnorm"
  noise_std: 0
  ckpt_path_vfgn: "models/new24-gpus-mean_me_final/model_loss-6.42E-04_step-350.pt"
  output_path: "rollouts/test2404"
  NUM_PARTICLE_TYPES: 3
  KINEMATIC_PARTICLE_ID: 0  # refers to anchor point
  METAL_PARTICLE_ID: 2  # refers to normal particles
  ANCHOR_PLANE_PARTICLE_ID: 1  # refers to anchor plane

test_options:
  rollout_refine: False  # Set False for: rollout the predictions, True for: single-step prediction for all steps
  rollout_path: "rollouts/test24/rollout_test_0.json"
  metadata_path: "./data/2024-3-absnorm"
  step_stride: 3
  block_on_show: True
  ds_type: str = "standard"  # test data type: ['standard', 'train', 'test']
  test_build_name: "test0"
  plot_tolerance_range: True
  plot_3d: False