不打印网络结构

2021-11-17 16:41:33 +08:00 · 2021-11-17 16:41:33 +08:00 · c2f0991139
parent cf8315c5b3
commit c2f0991139
1 changed files with 1 additions and 1 deletions
--- a/fastreid/engine/defaults.py
+++ b/fastreid/engine/defaults.py
@ -366,7 +366,7 @@ class DefaultTrainer(TrainerBase):
        """
        model = build_model(cfg)
        logger = logging.getLogger(__name__)
-        logger.info("Model:\n{}".format(model))
+        # logger.info("Model:\n{}".format(model))
        return model

    @classmethod