PyRetri/requirements.txt

# -*- coding: utf-8 -*-
numpy
torch==2.2.2
torchvision==0.18.1
scikit-learn
yacs
tqdm
upload 2020-04-02 14:00:49 +08:00			`# -- coding: utf-8 --`
			`numpy`
torch version fix - pip doesn't have 2.3.1 2024-12-25 12:05:03 +08:00			`torch==2.2.2`
WARNING Change torch and torchvision version - this has NOT been tested 2024-12-25 11:50:22 +08:00			`torchvision==0.18.1`
Depreciation Fix - replace sklearn with scikit-learn in requirements.txt and setup.py - change imports from torchvision.models.utils to torch.hub in builder.py and misc.py 2024-11-12 15:28:38 +08:00			`scikit-learn`
upload 2020-04-02 14:00:49 +08:00			`yacs`
			`tqdm`