Skip to content

Commit 5a7a0d5

Browse files
authored
Update README.md
1 parent 8ebf8f8 commit 5a7a0d5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ When launching a training, a link will be printed in the console. Click on it to
141141
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:
142142

143143
```python
144-
net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana')
144+
net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana', pretrained=True)
145145
```
146146
The training was done with a 50% scale and bilinear upsampling.
147147

0 commit comments

Comments
 (0)