diff --git a/args.py b/args.py index e92bb08..46dd57c 100644 --- a/args.py +++ b/args.py @@ -75,6 +75,3 @@ args.model_output_size = 1024 if args.use_sr else 512 if args.use_sr and args.use_interpolation: args.use_interpolation = False -# if args.use_sr and args.sr_x4: -# print('debug') -# args.cacher_quality = 100 diff --git a/main.py b/main.py index 9fcb36f..8f12a58 100644 --- a/main.py +++ b/main.py @@ -886,7 +886,7 @@ def main(): need_a_copy = True - postprocessed_image = model_output.copy() + postprocessed_image = model_output if args.perf == 'main': print("Compute", time.perf_counter() - tic) tic = time.perf_counter()