apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use default DAT params

This commit is contained in:
Sean Sube 2023-12-31 12:28:51 -06:00
parent 759db71ac7
commit ec79602a11
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/convert/upscaling/dat.py
View File

@ -28,13 +28,7 @@ def convert_upscaling_dat(
return return
model = DAT( model = DAT(
# TODO: params upscale=scale,
num_in_ch=3,
num_out_ch=3,
num_feat=64,
num_block=23,
num_grow_ch=32,
scale=scale,
) )
torch_model = torch.load(source, map_location=conversion.map_location) torch_model = torch.load(source, map_location=conversion.map_location)