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 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)
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)
