Update README.md

milesial · Aug 19, 2021 · 5a7a0d5 · 5a7a0d5
1 parent 8ebf8f8
commit 5a7a0d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -141,7 +141,7 @@ When launching a training, a link will be printed in the console. Click on it to
 A [pretrained model](https://github.com/milesial/Pytorch-UNet/releases/tag/v2.0) is available for the Carvana dataset. It can also be loaded from torch.hub:
 
 ```python
-net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana')
+net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana', pretrained=True)
 ```
 The training was done with a 50% scale and bilinear upsampling.