From 1a15109b5803a2b9468be062c880c6b87c1fdc64 Mon Sep 17 00:00:00 2001 From: LeoXing1996 Date: Wed, 18 May 2022 23:29:49 +0800 Subject: [PATCH] remove debug log --- mmgen/core/runners/fp16_utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mmgen/core/runners/fp16_utils.py b/mmgen/core/runners/fp16_utils.py index 2742c40c9..d9ed90f27 100644 --- a/mmgen/core/runners/fp16_utils.py +++ b/mmgen/core/runners/fp16_utils.py @@ -47,11 +47,9 @@ def nan_to_num(x, nan=0.0, posinf=None, neginf=None, *, out=None): Tensor: Output tensor. """ try: - print(torch.__version__, 'do not get error') return torch.nan_to_num( x, nan=nan, posinf=posinf, neginf=neginf, out=out) except AttributeError: - print(torch.__version__, 'get error') if not isinstance(x, torch.Tensor): raise TypeError( f'argument input (position 1) must be Tensor, not {type(x)}')