Ejemplos de next_frame_savp en Python

Ejemplo n.º 1

Archivo: next_frame_savp_test.py Proyecto: 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)

Ejemplo n.º 2

Archivo: next_frame_test.py Proyecto: 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)

Ejemplo n.º 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)

Ejemplo n.º 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)

Ejemplo n.º 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)

Ejemplo n.º 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)