def testEndToEnd3D(self):
images = tf.random_uniform([1, 299, 299, 299, 3])
hparams = revnet.revnet_base()
hparams.dim = '3d'
hparams.mode = tf.estimator.ModeKeys.TRAIN
logits = revnet.revnet(images, hparams)
self.assertEquals(logits.shape, [1, 1, 1, 1, 3328])
def testEndToEnd3D(self): images = tf.random_uniform([1, 299, 299, 299, 3]) hparams = revnet.revnet_base() hparams.dim = '3d' hparams.mode = tf.estimator.ModeKeys.TRAIN logits = revnet.revnet(images, hparams) self.assertEquals(logits.shape, [1, 1, 1, 1, 3328])