31 строка
484 B
Plaintext
31 строка
484 B
Plaintext
# --------- pytorch --------- #
|
|
torch==1.8.1
|
|
torchvision==0.9.1
|
|
pytorch-lightning>=1.2.10
|
|
torchmetrics
|
|
|
|
# --------- hydra --------- #
|
|
hydra-core==1.1.0.dev5
|
|
hydra-colorlog==1.1.0.dev1
|
|
hydra-optuna-sweeper==1.1.0.dev1
|
|
# hydra-ray-launcher==0.1.2
|
|
# hydra-submitit-launcher==1.1.0
|
|
|
|
# --------- loggers --------- #
|
|
wandb>=0.10.26
|
|
# neptune-client
|
|
# comet-ml
|
|
# mlflow
|
|
|
|
# --------- others --------- #
|
|
rich
|
|
python-dotenv
|
|
pytest
|
|
pre-commit
|
|
scikit-learn
|
|
pandas
|
|
matplotlib
|
|
seaborn
|
|
# pudb
|
|
# dvc
|