fix fp32 for shufflecugan

mafiosnik777 · Apr 24, 2023 · 944af83 · 944af83
1 parent 1918b13
commit 944af83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/env/inference/shufflecugan_ncnn.py b/src/env/inference/shufflecugan_ncnn.py
@@ -59,10 +59,10 @@ def execute(n, upscaled, metric_thresh, f):
 
 ncnn_shape_hint = (in_tile_channels, in_tile_height, in_tile_width, out_tile_channels, out_tile_height, out_tile_width)
 
-if tiling == False:
+if fp16 == False:
     upscaled = core.ncnn.Model(clip, network_path=engine, num_streams=threading(), fp16=True, use_ncnn_network_format=True, ncnn_shape_hint=ncnn_shape_hint)
 else:
-    upscaled = core.ncnn.Model(clip, network_path=engine, num_streams=threading(), fp16=False, use_ncnn_network_format=True, ncnn_shape_hint=ncnn_shape_hint, tilesize=[tileHeight, tileWidth])
+    upscaled = core.ncnn.Model(clip, network_path=engine, num_streams=threading(), fp16=False, use_ncnn_network_format=True, ncnn_shape_hint=ncnn_shape_hint)
 
 if frameskip:
     metric_thresh = 0.999