File size: 664 Bytes
8cc0674 |
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 |
# --------- 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 --------- # rootutils # standardizing the project root setup pre-commit # hooks for applying linters on commit rich # beautiful text formatting in terminal pytest # tests # sh # for running bash commands in some tests (linux/macos only) |