From b2553c09a723b3f68e80347b04ddfca1ed66b6cc Mon Sep 17 00:00:00 2001 From: Moritz Gunz Date: Fri, 5 Jul 2024 14:20:09 +0200 Subject: [PATCH] Torch: print model at log verbosity 3 Closes #1575 --- returnn/torch/engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/returnn/torch/engine.py b/returnn/torch/engine.py index 2b83c47e1..0558ba97e 100644 --- a/returnn/torch/engine.py +++ b/returnn/torch/engine.py @@ -862,7 +862,7 @@ def _create_model(self, *, epoch: int, step: int): else: raise TypeError(f"get_model returned {model} of type {type(model)}, expected rf.Module or torch.nn.Module") assert isinstance(self._pt_model, torch.nn.Module) - print("Model:", self._pt_model, file=log.v4) + print("Model:", self._pt_model, file=log.v3) num_params = sum([parameter.numel() for parameter in self._pt_model.parameters()]) print(f"net params #: {num_params}", file=log.v2)