Pay attention to declare the num classes as needed in my_utils.py and also in the train.py

Loading the data:

python ./utils/my_utils.py

training:

python ./train.py --config ./config.json

validating:

python ./test.py --resume ./saved/models/new_model/0211_113927/checkpoint-epoch30.pth --c ./config.json