mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 18:44:21 +08:00
changed test config file to inherit from default (#49)
* lint * reconfigured modules in yaml * added data module * changed test config file
This commit is contained in:
1 parent
d79c112248
commit
e874e856ec
4 files changed
+24
-7
No files matched your search
+1
-1
@@ -164,7 +164,7 @@ runs/
|
||||
outputs/
|
||||
|
||||
# Data
|
||||
data/
|
||||
./data/
|
||||
|
||||
# Misc
|
||||
old/
|
||||
|
||||
@@ -20,4 +20,4 @@ machine_config:
|
||||
num_gpus: 1
|
||||
num_machines: 1
|
||||
machine_rank: 0
|
||||
dist_url: auto
|
||||
dist_url: auto
|
||||
@@ -0,0 +1,18 @@
|
||||
defaults:
|
||||
- default
|
||||
|
||||
dry_run: true
|
||||
|
||||
machine_config:
|
||||
num_gpus: 0
|
||||
|
||||
data:
|
||||
dataset:
|
||||
data_directory: data/blender/lego_test
|
||||
validation_image_indices: [0]
|
||||
dataloader:
|
||||
num_images_to_sample_from: 1
|
||||
num_rays_per_batch: 8
|
||||
|
||||
graph:
|
||||
max_num_iterations: 5
|
||||
@@ -2,8 +2,7 @@
|
||||
Default test to make sure train runs
|
||||
"""
|
||||
import pytest
|
||||
import omegaconf
|
||||
from mattport.utils.io import get_absolute_path
|
||||
from hydra.experimental import compose, initialize
|
||||
|
||||
from scripts.run_train import main
|
||||
|
||||
@@ -11,6 +10,6 @@ from scripts.run_train import main
|
||||
@pytest.mark.filterwarnings("ignore::DeprecationWarning")
|
||||
def test_run_train():
|
||||
"""test run train script works properly"""
|
||||
test_config = get_absolute_path("./tests/configs/test_default.yml")
|
||||
cfg = omegaconf.OmegaConf.load(test_config)
|
||||
main(cfg)
|
||||
initialize(config_path="../configs/")
|
||||
conf = compose("test_dryrun.yaml")
|
||||
main(conf)
|
||||
Reference in new issue
Block a user