lint(api): remove some confusing log messages
This commit is contained in:
parent
972f9de2ef
commit
52065ef317
|
@ -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:
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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(
|
||||
|
|
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue