Skip to content
This repository has been archived by the owner on Nov 1, 2024. It is now read-only.

Commit

Permalink
Update torch version
Browse files Browse the repository at this point in the history
  • Loading branch information
@Adrien987k
Adrien987k committed Dec 4, 2022
1 parent a73f567 commit 4e12602
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 4e12602

Please sign in to comment.