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()
|