Update README.md

Former-commit-id: 5a7a0d5a57467e43e8d10c9b28a99b67e0936803
This commit is contained in:
milesial 2021-08-19 11:15:41 +02:00 committed by GitHub
parent 1f1967d271
commit 890719f4b7

View file

@ -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.