Commit 1c027be9 authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

fix missing config file in integration test

Summary: PathManager is the old version, therefore config files are not saved to Manifold

Reviewed By: sstsai-adl

Differential Revision: D26882435

fbshipit-source-id: 089c53db90080737b10bafe43d154d34221a7ce2
parent 2d4ebf8c
...@@ -11,10 +11,10 @@ from typing import Dict, Iterator ...@@ -11,10 +11,10 @@ from typing import Dict, Iterator
import detectron2.utils.comm as comm import detectron2.utils.comm as comm
import torch import torch
from d2go.config import CfgNode from d2go.config import CfgNode
from fvcore.common.file_io import PathManager from detectron2.utils.file_io import PathManager
from tabulate import tabulate from tabulate import tabulate
from .tensorboard_log_util import get_tensorboard_log_dir from .tensorboard_log_util import get_tensorboard_log_dir # noqa: forwarding
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -88,6 +88,7 @@ def dump_trained_model_configs( ...@@ -88,6 +88,7 @@ def dump_trained_model_configs(
config_file = os.path.join(trained_model_config_dir, "{}.yaml".format(name)) config_file = os.path.join(trained_model_config_dir, "{}.yaml".format(name))
trained_model_configs[name] = config_file trained_model_configs[name] = config_file
if comm.is_main_process(): if comm.is_main_process():
logger.info("Dump trained config file: {}".format(config_file))
with PathManager.open(config_file, "w") as f: with PathManager.open(config_file, "w") as f:
f.write(trained_cfg.dump()) f.write(trained_cfg.dump())
return trained_model_configs return trained_model_configs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment