Python unit Examples

Example #1

File: revnet_test.py Project: chqiwang/tensor2tensor

 def testUnit4(self):
   x1 = tf.random_uniform([1, 19, 19, 1024])
   x2 = tf.random_uniform([1, 19, 19, 1024])
   x1, x2 = revnet.unit(x1, x2, block_num=4, depth=416,
                        num_layers=1, stride=2)
   self.assertEquals(x1.get_shape().as_list(), [1, 10, 10, 1664])
   self.assertEquals(x2.get_shape().as_list(), [1, 10, 10, 1664])

Example #2

File: revnet_test.py Project: chqiwang/tensor2tensor

 def testUnit3D(self):
   x1 = tf.random_uniform([4, 74, 74, 74, 256])
   x2 = tf.random_uniform([4, 74, 74, 74, 256])
   x1, x2 = revnet.unit(x1, x2, block_num=5, depth=128,
                        num_layers=1, dim='3d', stride=2)
   self.assertEquals(x1.get_shape().as_list(), [4, 37, 37, 37, 512])
   self.assertEquals(x2.get_shape().as_list(), [4, 37, 37, 37, 512])

Example #3

File: revnet_test.py Project: chqiwang/tensor2tensor

 def testUnit1(self):
   x1 = tf.random_uniform([4, 74, 74, 256])
   x2 = tf.random_uniform([4, 74, 74, 256])
   x1, x2 = revnet.unit(x1, x2, block_num=1, depth=64,
                        first_batch_norm=True, num_layers=1)
   self.assertEquals(x1.get_shape().as_list(), [4, 74, 74, 256])
   self.assertEquals(x2.get_shape().as_list(), [4, 74, 74, 256])

Example #4

File: revnet_test.py Project: chqiwang/tensor2tensor

 def testUnit3(self):
   x1 = tf.random_uniform([1, 37, 37, 512])
   x2 = tf.random_uniform([1, 37, 37, 512])
   x1, x2 = revnet.unit(x1, x2, block_num=3, depth=256,
                        num_layers=10, stride=2)
   self.assertEquals(x1.get_shape().as_list(), [1, 19, 19, 1024])
   self.assertEquals(x2.get_shape().as_list(), [1, 19, 19, 1024])

Example #5

File: revnet_test.py Project: AranKomat/tensor2tensor

 def testUnit2(self):
   x1 = tf.random_uniform([4, 74, 74, 256])
   x2 = tf.random_uniform([4, 74, 74, 256])
   x1, x2 = revnet.unit(x1, x2, block_num=2, depth1=128, depth2=512,
                        num_layers=1, stride=2)
   self.assertEquals(x1.get_shape(), [4, 37, 37, 512])
   self.assertEquals(x2.get_shape(), [4, 37, 37, 512])

Example #6

File: revnet_test.py Project: Alwin-Bucher/tensor2tensor-1

 def testUnit3D(self):
   x1 = tf.random_uniform([4, 74, 74, 74, 256])
   x2 = tf.random_uniform([4, 74, 74, 74, 256])
   x1, x2 = revnet.unit(x1, x2, block_num=5, depth=128,
                        num_layers=1, dim='3d', stride=2)
   self.assertEqual(x1.get_shape().as_list(), [4, 37, 37, 37, 512])
   self.assertEqual(x2.get_shape().as_list(), [4, 37, 37, 37, 512])

Example #7

File: revnet_test.py Project: Alwin-Bucher/tensor2tensor-1

 def testUnit4(self):
   x1 = tf.random_uniform([1, 19, 19, 1024])
   x2 = tf.random_uniform([1, 19, 19, 1024])
   x1, x2 = revnet.unit(x1, x2, block_num=4, depth=416,
                        num_layers=1, stride=2)
   self.assertEqual(x1.get_shape().as_list(), [1, 10, 10, 1664])
   self.assertEqual(x2.get_shape().as_list(), [1, 10, 10, 1664])

Example #8

File: revnet_test.py Project: Alwin-Bucher/tensor2tensor-1

 def testUnit3(self):
   x1 = tf.random_uniform([1, 37, 37, 512])
   x2 = tf.random_uniform([1, 37, 37, 512])
   x1, x2 = revnet.unit(x1, x2, block_num=3, depth=256,
                        num_layers=10, stride=2)
   self.assertEqual(x1.get_shape().as_list(), [1, 19, 19, 1024])
   self.assertEqual(x2.get_shape().as_list(), [1, 19, 19, 1024])

Example #9

File: revnet_test.py Project: Alwin-Bucher/tensor2tensor-1

 def testUnit1(self):
   x1 = tf.random_uniform([4, 74, 74, 256])
   x2 = tf.random_uniform([4, 74, 74, 256])
   x1, x2 = revnet.unit(x1, x2, block_num=1, depth=64,
                        first_batch_norm=True, num_layers=1)
   self.assertEqual(x1.get_shape().as_list(), [4, 74, 74, 256])
   self.assertEqual(x2.get_shape().as_list(), [4, 74, 74, 256])

Example #10

File: revnet_test.py Project: yzx1992/tensor2tensor

 def testUnit2(self):
     x1 = tf.random_uniform([4, 74, 74, 256])
     x2 = tf.random_uniform([4, 74, 74, 256])
     x1, x2 = revnet.unit(x1,
                          x2,
                          block_num=2,
                          depth1=128,
                          depth2=512,
                          num_layers=1,
                          stride=2)
     self.assertEquals(x1.get_shape(), [4, 37, 37, 512])
     self.assertEquals(x2.get_shape(), [4, 37, 37, 512])