Ejemplo n.º 1
0
 def testSummary(self):
     with self.test_session():
         inputs = tf.constant(_rand(1, 18, 19, 5))
         spec = "net = Cr(64, [5, 5])"
         outputs = specs.create_net(spec, inputs)
         tf.initialize_all_variables().run()
         result = outputs.eval()
         self.assertEqual(tuple(result.shape), (1, 18, 19, 64))
         summaries.tf_spec_summary(spec, inputs)
Ejemplo n.º 2
0
 def testSummary(self):
   with self.test_session():
     inputs = tf.constant(_rand(1, 18, 19, 5))
     spec = "net = Cr(64, [5, 5])"
     outputs = specs.create_net(spec, inputs)
     tf.global_variables_initializer().run()
     result = outputs.eval()
     self.assertEqual(tuple(result.shape), (1, 18, 19, 64))
     summaries.tf_spec_summary(spec, inputs)