There is no PyPI repo. Install from github:

pip install -U git+https://github.com/ihoromi4/bengali-ai.git

• pytorch
• torchvision
• pretrainedmodels
• catalyst
• albumentations
• efficientnet_pytorch
• iterative-stratification
• wandb

I used W&B service for logging and experiment analysis. So you need W&B account to run experiments.

1. Download competition dataset (bash):

pip install kaggle
kaggle competitions download -c bengaliai-cv19
unzip -u bengaliai-cv19.zip

2. Run experiment (python):

from bengaliai.experiments.seresnext50_deephead import experiment

experiment.run(check=False)