Python next_frame_savp Examples

Example #1

File: next_frame_savp_test.py Project: latifisalar/tensor2tensor

 def testStochasticSavp(self):
   self.TestOnVariousInputOutputSizes(
       next_frame_params.next_frame_savp(),
       next_frame_savp.NextFrameSAVP,
       1)
   self.TestOnVariousUpSampleLayers(
       next_frame_params.next_frame_savp(),
       next_frame_savp.NextFrameSAVP,
       1)

Example #2

File: next_frame_test.py Project: yufengm/tensor2tensor

 def testStochasticInvalidVAEGANCombinations(self):
     hparams = next_frame_params.next_frame_savp()
     for use_vae, use_gan in [[True, True], [False, False]]:
         hparams.use_gan = use_gan
         hparams.use_vae = use_vae
         self.assertRaises(ValueError, self.TestVideoModel, 7, 5, hparams,
                           next_frame_savp.NextFrameSAVP, 1)

Example #3

    def testStochasticSavpGAN(self):
        hparams = next_frame_params.next_frame_savp()
        hparams.use_gan = True
        hparams.use_vae = False
        self.TestVideoModel(7, 5, hparams, next_frame_savp.NextFrameSAVP, 1)

        hparams.gan_optimization = "sequential"
        self.TestVideoModel(7, 5, hparams, next_frame_savp.NextFrameSAVP, 1)

Example #4

 def testStochasticSavpVAE(self):
     savp_hparams = next_frame_params.next_frame_savp()
     savp_hparams.use_vae = True
     savp_hparams.use_gan = False
     self.TestOnVariousInputOutputSizes(savp_hparams,
                                        next_frame_savp.NextFrameSAVP, 1)
     self.TestOnVariousUpSampleLayers(savp_hparams,
                                      next_frame_savp.NextFrameSAVP, 1)

Example #5

 def testStochasticInvalidVAEGANCombinations(self):
     hparams = next_frame_params.next_frame_savp()
     hparams.use_gan = False
     hparams.use_vae = False
     self.assertRaises(ValueError, self.TestVideoModel, 7, 5, hparams,
                       next_frame_savp.NextFrameSAVP, 1)

Example #6

 def testStochasticSavpGANVAE(self):
     hparams = next_frame_params.next_frame_savp()
     hparams.use_vae = True
     hparams.use_gan = True
     self.TestVideoModel(7, 5, hparams, next_frame_savp.NextFrameSAVP, 1)