| # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. 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. | |
| config_file=configs/accelerate/accelerate_config.yaml | |
| train_file=train.py | |
| ### Static 3D Generation ### | |
| # Stage 1 | |
| config1=configs/training/3dgs_res_176_320_views_17.yaml | |
| # accelerate launch --config_file $config_file $train_file --config $config1 | |
| # Stage 2 | |
| config2=configs/training/3dgs_res_176_320_views_49.yaml | |
| python src/utils/copy_to_resume.py $config1 $config2 | |
| # accelerate launch --config_file $config_file $train_file --config $config2 | |
| # Stage 3 | |
| config3=configs/training/3dgs_res_352_640_views_49.yaml | |
| python src/utils/copy_to_resume.py $config2 $config3 | |
| # accelerate launch --config_file $config_file $train_file --config $config3 | |
| # Stage 4 | |
| config4=configs/training/3dgs_res_704_1280_views_49.yaml | |
| python src/utils/copy_to_resume.py $config3 $config4 | |
| # accelerate launch --config_file $config_file $train_file --config $config4 | |
| # Stage 5 | |
| config5=configs/training/3dgs_res_704_1280_views_121.yaml | |
| python src/utils/copy_to_resume.py $config4 $config5 | |
| # accelerate launch --config_file $config_file $train_file --config $config5 | |
| # Stage 6 | |
| config6=configs/training/3dgs_res_704_1280_views_121_multi_6.yaml | |
| python src/utils/copy_to_resume.py $config5 $config6 | |
| # accelerate launch --config_file $config_file $train_file --config $config6 | |
| # Stage 7 | |
| config7=configs/training/3dgs_res_704_1280_views_121_multi_6_prune.yaml | |
| python src/utils/copy_to_resume.py $config6 $config7 | |
| # accelerate launch --config_file $config_file $train_file --config $config7 | |
| ### Dynamic 3D Generation ### | |
| # Stage 8 | |
| config8=configs/training/3dgs_res_704_1280_views_121_multi_6_dynamic.yaml | |
| python src/utils/copy_to_resume.py $config6 $config8 | |
| accelerate launch --config_file $config_file $train_file --config $config8 | |
| config9=configs/training/3dgs_res_704_1280_views_121_multi_6_dynamic_prune.yaml | |
| python src/utils/copy_to_resume.py $config8 $config9 | |
| accelerate launch --config_file $config_file $train_file --config $config9 | |