def testSavpGAN(self): hparams = savp_params.next_frame_savp() hparams.use_gan = True hparams.use_vae = False self.TestVideoModel(7, 5, hparams, savp.NextFrameSAVP, 1) hparams.gan_optimization = "sequential" self.TestVideoModel(7, 5, hparams, savp.NextFrameSAVP, 1)
def testSavpVAE(self): savp_hparams = savp_params.next_frame_savp() savp_hparams.use_vae = True savp_hparams.use_gan = False self.TestOnVariousInputOutputSizes( savp_hparams, savp.NextFrameSAVP, 1) self.TestOnVariousUpSampleLayers( savp_hparams, savp.NextFrameSAVP, 1)
def testInvalidVAEGANCombinations(self): hparams = savp_params.next_frame_savp() hparams.use_gan = False hparams.use_vae = False self.assertRaises(ValueError, self.TestVideoModel, 7, 5, hparams, savp.NextFrameSAVP, 1)
def testSavpGANVAE(self): hparams = savp_params.next_frame_savp() hparams.use_vae = True hparams.use_gan = True self.TestVideoModel(7, 5, hparams, savp.NextFrameSAVP, 1)