From 4e12602fd495af83efd1631fbe82523e6db092e0 Mon Sep 17 00:00:00 2001 From: Adrien Bardes Date: Sun, 4 Dec 2022 09:49:22 -0800 Subject: [PATCH] Update torch version --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 4c354a14..c17da395 100644 --- a/README.md +++ b/README.md @@ -62,7 +62,7 @@ resnet200x2 = torch.hub.load('facebookresearch/vicreg:main', 'resnet200x2') ## Training -Install PyTorch ([pytorch.org](http://pytorch.org)) and download [ImageNet](https://imagenet.stanford.edu/). The code has been developed for PyTorch version 1.7.1 and torchvision version 0.8.2, but should work with other versions just as well. +Install PyTorch ([pytorch.org](http://pytorch.org)) and download [ImageNet](https://imagenet.stanford.edu/). The code has been developed for PyTorch version 1.8.1 and torchvision version 0.9.1, but should work with other versions just as well. ### Single-node local training