xla

c7b87dce - add a --model flag to use any (supported) torchvision model

add a --model flag to use any (supported) torchvision model tested as: ```bash python $code_source/test/test_train_imagenet.py \ --datadir='$data_source' --model='resnet50' ``` for now only supported model is resnet50