From 890719f4b7592775437406879ffac5fb56eef1e6 Mon Sep 17 00:00:00 2001
From: milesial <milesial@users.noreply.github.com>
Date: Thu, 19 Aug 2021 11:15:41 +0200
Subject: [PATCH] Update README.md

Former-commit-id: 5a7a0d5a57467e43e8d10c9b28a99b67e0936803
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index d2a4baf..3620a45 100644
--- 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.