ssl-in-the-wild/requirements.txt at main · HLCaptain/ssl-in-the-wild · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# --------- ssl --------- #
lightly>=1.4.21

# --------- pytorch --------- #
torch>=2.0.0
torchvision>=0.15.0
lightning>=2.0.0
torchmetrics>=0.11.4

# --------- hydra --------- #
hydra-core==1.3.2
hydra-colorlog==1.2.0
hydra-optuna-sweeper==1.2.0

# --------- loggers --------- #
wandb
# neptune-client
# mlflow
# comet-ml
# aim>=3.16.2  # no lower than 3.16.2, see https://github.com/aimhubio/aim/issues/2550

# --------- others --------- #
traitlets==5.9.0
rootutils       # standardizing the project root setup
pre-commit      # hooks for applying linters on commit
rich            # beautiful text formatting in terminal
pytest          # tests
chardet		# character encoding detector
types-chardet	# type stub package for the chardet package
# sh            # for running bash commands in some tests (linux/macos only)

# --------- load data --------- #
opendatasets>=0.1.22 # downloading datasets from online sources like Kaggle

# --------- visualization --------- #
pandas>=1.5.3 # data structures and data analysis tools
matplotlib>=3.8.0