Skip to content

Commit 8ebf8f8

Browse files
authored
Update README
1 parent 7a89332 commit 8ebf8f8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,12 +138,12 @@ When launching a training, a link will be printed in the console. Click on it to
138138

139139

140140
## Pretrained model
141-
A [pretrained model](https://github.com/milesial/Pytorch-UNet/releases/tag/v1.0) is available for the Carvana dataset. It can also be loaded from torch.hub:
141+
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
144144
net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana')
145145
```
146-
The training was done with a 100% scale and bilinear upsampling.
146+
The training was done with a 50% scale and bilinear upsampling.
147147

148148
## Data
149149
The Carvana data is available on the [Kaggle website](https://www.kaggle.com/c/carvana-image-masking-challenge/data).

0 commit comments

Comments
 (0)