From 1f1967d27167a370933c6d547fffae431572d973 Mon Sep 17 00:00:00 2001 From: milesial Date: Thu, 19 Aug 2021 11:14:36 +0200 Subject: [PATCH] Update README Former-commit-id: 8ebf8f874be976cace2d32365c8dce31fe7d6d5a --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 1a3b413..d2a4baf 100644 --- a/README.md +++ b/README.md @@ -138,12 +138,12 @@ When launching a training, a link will be printed in the console. Click on it to ## Pretrained model -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: +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') ``` -The training was done with a 100% scale and bilinear upsampling. +The training was done with a 50% scale and bilinear upsampling. ## Data The Carvana data is available on the [Kaggle website](https://www.kaggle.com/c/carvana-image-masking-challenge/data).