mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-09 15:02:03 +00:00
Update README.md
Former-commit-id: 5a7a0d5a57467e43e8d10c9b28a99b67e0936803
This commit is contained in:
parent
1f1967d271
commit
890719f4b7
|
@ -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.
|
||||
|
||||
|
|
Loading…
Reference in a new issue