52Hz commited on
Commit
f9999de
1 Parent(s): 3f8d0fd

Update main_test_SRMNet.py

Browse files
Files changed (1) hide show
  1. main_test_SRMNet.py +0 -5
main_test_SRMNet.py CHANGED
@@ -81,11 +81,6 @@ def load_checkpoint(model, weights):
81
  new_state_dict[name] = v
82
  model.load_state_dict(new_state_dict)
83
 
84
- def setup(args):
85
- save_dir = 'result/'
86
- folder = 'test/'
87
-
88
- return folder, save_dir
89
 
90
  if __name__ == '__main__':
91
  main()
81
  new_state_dict[name] = v
82
  model.load_state_dict(new_state_dict)
83
 
 
 
 
 
 
84
 
85
  if __name__ == '__main__':
86
  main()