1
0
Fork 0

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

View File

@ -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)