XiangZ
/

hit-sr

XiangZ commited on May 13

Commit

58d1c2c

verified ·

1 Parent(s): 2e01b14

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -62,7 +62,8 @@ model = model.from_pretrained(repo_name)
 model.to(device)
 # test and save results
-sr_results = model.infer_image("path-to-input-image", device=device)
 cv2.imwrite("path-to-output-location", sr_results)
 ```

 model.to(device)
 # test and save results
+with torch.no_grad():
+    sr_results = model.infer_image("path-to-input-image", device=device)
 cv2.imwrite("path-to-output-location", sr_results)
 ```