PyRetri/requirements.txt

mirror of https://github.com/PyRetri/PyRetri.git
# -*- coding: utf-8 -*-
numpy
torch>=1.1
torchvision>=0.4
scikit-learn
yacs
tqdm
upload 2020-04-02 14:00:49 +08:00			`# -- coding: utf-8 --`
			`numpy`
			`torch>=1.1`
			`torchvision>=0.4`
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`