"""Store for caffe2 examples and common models."""from__future__importabsolute_importas_absimportosimportsysimportimportlibfrom.importsqueezenetfromcaffe2.python.models.downloadimportModelDownloadermodels=['squeezenet','resnet50','vgg19',]mf=ModelDownloader()classModel:def__init__(self,model_name):self.init_net,self.predict_net,self.value_info=mf.get_c2_model(model_name)formodelinmodels:try:locals()['c2_'+model]=importlib.import_module('caffe2.python.models.'+model)exceptImportError:locals()['c2_'+model]=Model(model)# squeezenetdefrelay_squeezenet():returnsqueezenet.get_workload()