mirror of
https://github.com/PyRetri/PyRetri.git
synced 2025-06-03 14:49:50 +08:00
3043758bfd
- 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
8 lines
81 B
Plaintext
8 lines
81 B
Plaintext
# -*- coding: utf-8 -*-
|
|
numpy
|
|
torch>=1.1
|
|
torchvision>=0.4
|
|
scikit-learn
|
|
yacs
|
|
tqdm
|