From 52065ef3175ee371d0c4fb3f646f957ca4a1979b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 23 Dec 2023 12:17:23 -0600 Subject: [PATCH] lint(api): remove some confusing log messages --- api/onnx_web/convert/correction/gfpgan.py | 2 -- api/onnx_web/convert/upscaling/bsrgan.py | 1 - api/onnx_web/convert/upscaling/resrgan.py | 2 -- api/onnx_web/convert/upscaling/swinir.py | 1 - 4 files changed, 6 deletions(-) diff --git a/api/onnx_web/convert/correction/gfpgan.py b/api/onnx_web/convert/correction/gfpgan.py index 0662754b..5a5e9508 100644 --- a/api/onnx_web/convert/correction/gfpgan.py +++ b/api/onnx_web/convert/correction/gfpgan.py @@ -27,7 +27,6 @@ def convert_correction_gfpgan( logger.info("ONNX model already exists, skipping") return - logger.info("loading and training model") model = RRDBNet( num_in_ch=3, num_out_ch=3, @@ -38,7 +37,6 @@ def convert_correction_gfpgan( ) torch_model = torch.load(source, map_location=conversion.map_location) - # TODO: make sure strict=False is safe here if "params_ema" in torch_model: model.load_state_dict(torch_model["params_ema"], strict=False) else: diff --git a/api/onnx_web/convert/upscaling/bsrgan.py b/api/onnx_web/convert/upscaling/bsrgan.py index 448a347a..915e4b9d 100644 --- a/api/onnx_web/convert/upscaling/bsrgan.py +++ b/api/onnx_web/convert/upscaling/bsrgan.py @@ -27,7 +27,6 @@ def convert_upscaling_bsrgan( logger.info("ONNX model already exists, skipping") return - logger.info("loading and training model") # values based on https://github.com/cszn/BSRGAN/blob/main/main_test_bsrgan.py#L69 model = RRDBNet( num_in_ch=3, diff --git a/api/onnx_web/convert/upscaling/resrgan.py b/api/onnx_web/convert/upscaling/resrgan.py index d5ac359d..291670a6 100644 --- a/api/onnx_web/convert/upscaling/resrgan.py +++ b/api/onnx_web/convert/upscaling/resrgan.py @@ -31,8 +31,6 @@ def convert_upscale_resrgan( logger.info("ONNX model already exists, skipping") return - logger.info("loading and training model") - if TAG_X4_V3 in name: # the x4-v3 model needs a different network model = SRVGGNetCompact( diff --git a/api/onnx_web/convert/upscaling/swinir.py b/api/onnx_web/convert/upscaling/swinir.py index 36b6168c..d6add243 100644 --- a/api/onnx_web/convert/upscaling/swinir.py +++ b/api/onnx_web/convert/upscaling/swinir.py @@ -27,7 +27,6 @@ def convert_upscaling_swinir( logger.info("ONNX model already exists, skipping") return - logger.info("loading and training model") # values based on https://github.com/JingyunLiang/SwinIR/blob/main/main_test_swinir.py#L128 params = { "depths": [6] * 6,