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)