Caltech101 dataset will give 404 error when downloading via torchvision, thus I have replaced it with CIFAR10
merged the train_with_engine and train_with_trainer scripts
added barrier context for data downloading so that process 0 will download and other processes will wait for the downloading to finish before moving on, this can avoid data corruption.
train_with_engine
andtrain_with_trainer
scripts