download_and_train does not train the model

[Surabhiad]

Description

it takes just a few seconds to run download_and_train.py. Also, when I run predict.py it takes just a few seconds. predicted images aren't saved either. so, I don't think the model is being trained. it doesn't show any errors though. but I am not being able to train or predict.

Expected Behavior

download_and_train.py trains