update

src/utils/misc.py

@@ -1,4 +1,3 @@
-from loguru import logger
 import torch
 from yacs.config import CfgNode as CN
 
@@ -17,9 +16,5 @@ def upper_config(dict_cfg):
 
 
 def detect_NaN(feat_0, feat_1):
-    logger.info(f"NaN detected in feature")
-    logger.info(
-        f"#NaN in feat_0: {torch.isnan(feat_0).int().sum()}, #NaN in feat_1: {torch.isnan(feat_1).int().sum()}"
-    )
     feat_0[torch.isnan(feat_0)] = 0
     feat_1[torch.isnan(feat_1)] = 0