modified: EI_mxnet_module_API_Resnet50.py

awslabs · Jul 31, 2019 · 492bad2 · 492bad2
1 parent ceceaff
commit 492bad2
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ec2-spot-elastic-inference/EI_mxnet_module_API_Resnet50.py b/ec2-spot-elastic-inference/EI_mxnet_module_API_Resnet50.py
@@ -15,13 +15,13 @@
 
 image = args.image
 
-#path='http://data.mxnet.io/models/imagenet/'
+path='http://data.mxnet.io/models/imagenet/'
 
-#[mx.test_utils.download(path+'resnet/50-layers/resnet-50-0000.params'),
+[mx.test_utils.download(path+'resnet/50-layers/resnet-50-0000.params'),
 
-#mx.test_utils.download(path+'resnet/50-layers/resnet-50-symbol.json'),
+mx.test_utils.download(path+'resnet/50-layers/resnet-50-symbol.json'),
 
-#mx.test_utils.download(path+'synset.txt')]
+mx.test_utils.download(path+'synset.txt')]
 
 
 path = 'http://data.dmlc.ml/models/imagenet/squeezenet/'
@@ -88,4 +88,4 @@
 for i in a[0:5]:
 
   print('probability=%f, class=%s' %(prob[i], labels[i]))
-
+