diff --git a/timm_image.py b/timm_image.py index e3ce234..45dca20 100644 --- a/timm_image.py +++ b/timm_image.py @@ -38,6 +38,7 @@ import warnings warnings.filterwarnings('ignore') log = logging.getLogger('timm_op') +log.setLevel(logging.ERROR) def torch_no_grad(f):