(Minor) Remove cli_from_base_configs from view_dataset.py (#376)

Remove cli_from_base_configs from view_dataset.py
This commit is contained in:
Brent Yi authored and GitHub committed 2022-09-08 19:48:09 -07:00
1 parent 97b77c6c5f
commit 12958ce1c3
1 file changed
+3 -2
+3 -2
View File
@@ -5,12 +5,12 @@ view_dataset.py
import logging
from datetime import timedelta
import dcargs
import torch
import yaml
from rich.console import Console
from nerfactory.configs import base as cfg
from nerfactory.configs.utils import cli_from_base_configs
from nerfactory.viewer.server import viewer_utils
logging.basicConfig(format="[%(filename)s:%(lineno)d] %(message)s", level=logging.DEBUG)
@@ -36,7 +36,8 @@ if __name__ == "__main__":
from nerfactory.configs.base_configs import base_configs
instantiated_config = cli_from_base_configs(base_configs)
SubcommandType = dcargs.extras.subcommand_type_from_defaults(base_configs)
instantiated_config = dcargs.cli(SubcommandType)
if instantiated_config.trainer.load_config:
logging.info(f"Loading pre-set config from: {instantiated_config.trainer.load_config}")
instantiated_config = yaml.load(instantiated_config.trainer.load_config.read_text(), Loader=yaml.Loader)