Debug main_test_swin2sr.py

mv-lab · Sep 29, 2022 · cacccc7 · cacccc7
1 parent 8f9aa49
commit cacccc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main_test_swin2sr.py b/main_test_swin2sr.py
@@ -191,7 +191,7 @@ def define_model(args):
 def setup(args):
  # 001 classical image sr/ 002 lightweight image sr
  if args.task in ['classical_sr', 'lightweight_sr', 'compressed_sr']:
- save_dir = f'results/swinir_{args.task}_x{args.scale}'
+ save_dir = f'results/swin2sr_{args.task}_x{args.scale}'
  folder = args.folder_gt
  border = args.scale
  window_size = 8