@@ -807,7 +807,7 @@ def test_caffe2_simple_model(self):
807807 model = ModelHelper (name = "mnist" )
808808 # how come those inputs don't break the forward pass =.=a
809809 workspace .FeedBlob ("data" , np .random .randn (1 , 3 , 64 , 64 ).astype (np .float32 ))
810- workspace .FeedBlob ("label" , np .random .randn (1 , 1000 ).astype (np . int ))
810+ workspace .FeedBlob ("label" , np .random .randn (1 , 1000 ).astype (int ))
811811
812812 with core .NameScope ("conv1" ):
813813 conv1 = brew .conv (model , "data" , 'conv1' , dim_in = 1 , dim_out = 20 , kernel = 5 )
@@ -842,7 +842,7 @@ def test_caffe2_simple_model(self):
842842 def test_caffe2_simple_cnnmodel (self ):
843843 model = cnn .CNNModelHelper ("NCHW" , name = "overfeat" )
844844 workspace .FeedBlob ("data" , np .random .randn (1 , 3 , 64 , 64 ).astype (np .float32 ))
845- workspace .FeedBlob ("label" , np .random .randn (1 , 1000 ).astype (np . int ))
845+ workspace .FeedBlob ("label" , np .random .randn (1 , 1000 ).astype (int ))
846846 with core .NameScope ("conv1" ):
847847 conv1 = model .Conv ("data" , "conv1" , 3 , 96 , 11 , stride = 4 )
848848 relu1 = model .Relu (conv1 , conv1 )
0 commit comments