From ec79602a11785473180eb12175573dca8a8d6d9d Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 31 Dec 2023 12:28:51 -0600 Subject: [PATCH] use default DAT params --- api/onnx_web/convert/upscaling/dat.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/api/onnx_web/convert/upscaling/dat.py b/api/onnx_web/convert/upscaling/dat.py index 8259d62d..fdf5fa95 100644 --- a/api/onnx_web/convert/upscaling/dat.py +++ b/api/onnx_web/convert/upscaling/dat.py @@ -28,13 +28,7 @@ def convert_upscaling_dat( return model = DAT( - # TODO: params - num_in_ch=3, - num_out_ch=3, - num_feat=64, - num_block=23, - num_grow_ch=32, - scale=scale, + upscale=scale, ) torch_model = torch.load(source, map_location=conversion.map_location)