diff --git a/examples/cfd/external_aerodynamics/figconvnet/README.md b/examples/cfd/external_aerodynamics/figconvnet/README.md index 96d51cef5f..c7a549e9ae 100644 --- a/examples/cfd/external_aerodynamics/figconvnet/README.md +++ b/examples/cfd/external_aerodynamics/figconvnet/README.md @@ -49,7 +49,7 @@ The binary dataset should have the following structure: ```text ├─ partitions │ ├─ graph_partitions_1.bin -│ ├─ graph_partitions_1.bin +│ ├─ graph_partitions_2.bin │ ├─ ... ├─ test_partitions │ ├─ graph_partitions_100.bin diff --git a/examples/cfd/external_aerodynamics/figconvnet/src/data/drivaerml_datamodule.py b/examples/cfd/external_aerodynamics/figconvnet/src/data/drivaerml_datamodule.py index c18ba85962..1cc0902273 100644 --- a/examples/cfd/external_aerodynamics/figconvnet/src/data/drivaerml_datamodule.py +++ b/examples/cfd/external_aerodynamics/figconvnet/src/data/drivaerml_datamodule.py @@ -43,7 +43,7 @@ class DrivAerMLPartitionedDataset(Dataset): ``` ├─ partitions │ ├─ graph_partitions_1.bin - │ ├─ graph_partitions_1.bin + │ ├─ graph_partitions_2.bin │ ├─ ... ├─ test_partitions │ ├─ graph_partitions_100.bin diff --git a/examples/cfd/figconvnet/configs/experiment/drivaerml/figconv_unet.yaml b/examples/cfd/figconvnet/configs/experiment/drivaerml/figconv_unet.yaml deleted file mode 100644 index e3c25820d2..0000000000 --- a/examples/cfd/figconvnet/configs/experiment/drivaerml/figconv_unet.yaml +++ /dev/null @@ -1,58 +0,0 @@ -# @package _global_ - -# SPDX-FileCopyrightText: Copyright (c) 2023 - 2024 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. - -defaults: - - /data: drivaerml - - /model: figconv_unet_drivaerml - - /loss: mseloss - - /optimizer: adam - - /lr_scheduler: steplr - -train: - batch_size: 8 - num_epochs: 200 - -model: - aabb_max: [ 2.0, 1.8, 2.6] - aabb_min: [-2.0, -1.8, -1.5] - hidden_channels: [16, 16, 16] - kernel_size: 5 - # mlp_channels: [2048, 2048] - neighbor_search_type: radius - num_down_blocks: 1 - num_levels: 2 - pooling_layers: [2] - pooling_type: max - reductions: [mean] - resolution_memory_format_pairs: - - ${res_mem_pair:b_xc_y_z, [ 5, 150, 100]} - - ${res_mem_pair:b_yc_x_z, [250, 3, 100]} - - ${res_mem_pair:b_zc_x_y, [250, 150, 2]} - use_rel_pos_encode: true - -lr_scheduler: - step_size: 50 - -loggers: - wandb: - entity: modulus - project_name: car-cfd - group_name: fignet-drivaerml - run_name: FIGConvNet-level2-16,16,16-res250-150-100-pool-max-2-aabb-275x15x1-ks5-np32768-b8x2 - -seed: 0