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")
|
logger.info("ONNX model already exists, skipping")
|
||||||
return
|
return
|
||||||
|
|
||||||
logger.info("loading and training model")
|
|
||||||
model = RRDBNet(
|
model = RRDBNet(
|
||||||
num_in_ch=3,
|
num_in_ch=3,
|
||||||
num_out_ch=3,
|
num_out_ch=3,
|
||||||
|
@ -38,7 +37,6 @@ def convert_correction_gfpgan(
|
||||||
)
|
)
|
||||||
|
|
||||||
torch_model = torch.load(source, map_location=conversion.map_location)
|
torch_model = torch.load(source, map_location=conversion.map_location)
|
||||||
# TODO: make sure strict=False is safe here
|
|
||||||
if "params_ema" in torch_model:
|
if "params_ema" in torch_model:
|
||||||
model.load_state_dict(torch_model["params_ema"], strict=False)
|
model.load_state_dict(torch_model["params_ema"], strict=False)
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -27,7 +27,6 @@ def convert_upscaling_bsrgan(
|
||||||
logger.info("ONNX model already exists, skipping")
|
logger.info("ONNX model already exists, skipping")
|
||||||
return
|
return
|
||||||
|
|
||||||
logger.info("loading and training model")
|
|
||||||
# values based on https://github.com/cszn/BSRGAN/blob/main/main_test_bsrgan.py#L69
|
# values based on https://github.com/cszn/BSRGAN/blob/main/main_test_bsrgan.py#L69
|
||||||
model = RRDBNet(
|
model = RRDBNet(
|
||||||
num_in_ch=3,
|
num_in_ch=3,
|
||||||
|
|
|
@ -31,8 +31,6 @@ def convert_upscale_resrgan(
|
||||||
logger.info("ONNX model already exists, skipping")
|
logger.info("ONNX model already exists, skipping")
|
||||||
return
|
return
|
||||||
|
|
||||||
logger.info("loading and training model")
|
|
||||||
|
|
||||||
if TAG_X4_V3 in name:
|
if TAG_X4_V3 in name:
|
||||||
# the x4-v3 model needs a different network
|
# the x4-v3 model needs a different network
|
||||||
model = SRVGGNetCompact(
|
model = SRVGGNetCompact(
|
||||||
|
|
|
@ -27,7 +27,6 @@ def convert_upscaling_swinir(
|
||||||
logger.info("ONNX model already exists, skipping")
|
logger.info("ONNX model already exists, skipping")
|
||||||
return
|
return
|
||||||
|
|
||||||
logger.info("loading and training model")
|
|
||||||
# values based on https://github.com/JingyunLiang/SwinIR/blob/main/main_test_swinir.py#L128
|
# values based on https://github.com/JingyunLiang/SwinIR/blob/main/main_test_swinir.py#L128
|
||||||
params = {
|
params = {
|
||||||
"depths": [6] * 6,
|
"depths": [6] * 6,
|
||||||
|
|
Loading…
Reference in New Issue