To operate on this dataset, please make sure the dataset 'Animals with Attributes' dataset is available. In order to Download it, please run the follwing command
python prepare_dataset.py
The dataset is downloaded to path: /tmp/ (for linux)
In order to run train.py, make sure to run the requirements file
pip install -r requirements.txt