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