Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -24,7 +24,7 @@ def main():
     parser = argparse.ArgumentParser(description='Quick demo Image Restoration')
     parser.add_argument('--input_dir', default='test/', type=str, help='Input images root')
     parser.add_argument('--result_dir', default='result/', type=str, help='Results images root')
-    parser.add_argument('--weights_root', default='pretrained_model', type=str, help='Weights root')
+    parser.add_argument('--weights_root', default='experiments/pretrained_models', type=str, help='Weights root')
     parser.add_argument('--task', default='Retouching', type=str, help='Restoration task (Above task list)')
 
     args = parser.parse_args()