| # --------- 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) |
|
|
