def testSavpGAN(self): hparams = next_frame_savp_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)
def testSavpVAE(self): savp_hparams = next_frame_savp_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)
def testInvalidVAEGANCombinations(self): hparams = next_frame_savp_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)
def testSavpGANVAE(self): hparams = next_frame_savp_params.next_frame_savp() hparams.use_vae = True hparams.use_gan = True self.TestVideoModel(7, 5, hparams, next_frame_savp.NextFrameSAVP, 1)