Spaces:
Runtime error
Runtime error
Update main_test_SRMNet.py
Browse files- 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()
|