File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff line change 18
18
import paddle .v2 .data_type as data_type
19
19
import paddle .v2 .layer as layer
20
20
import paddle .v2 .pooling as pooling
21
+ import paddle .v2 .networks as networks
21
22
22
23
pixel = layer .data (name = 'pixel' , type = data_type .dense_vector (128 ))
23
24
label = layer .data (name = 'label' , type = data_type .integer_value (10 ))
@@ -251,5 +252,13 @@ def test_operator(self):
251
252
print layer .parse_network (conv1 )
252
253
253
254
255
+ class NetworkTests (unittest .TestCase ):
256
+ def test_vgg (self ):
257
+ img = layer .data (name = 'pixel' , type = data_type .dense_vector (784 ))
258
+ vgg_out = networks .small_vgg (
259
+ input_image = img , num_channels = 1 , num_classes = 2 )
260
+ print layer .parse_network (vgg_out )
261
+
262
+
254
263
if __name__ == '__main__' :
255
264
unittest .main ()
You can’t perform that action at this time.
0 commit comments