Deployed 0ce180f with MkDocs version: 1.0.4

2026-09-09 11:32:07 +08:00 · 2019-07-26 22:24:14 -05:00
parent 0985c7a1ca
commit 3007636435
5 changed files with 44 additions and 44 deletions
@@ -579,7 +579,8 @@ The meta_tags.csv file can be found in the test-tube experiment save_dir.
    map_location=None
 )

-# predict 
+# predict
+pretrained_model.eval()
 pretrained_model.freeze()
 y_hat = pretrained_model(x)
 </code></pre>