def test_one_epoch_without_function(self): epochs = 1 batch_size = 1 enable_function = False input_image = tf.random.uniform((28, 28, 1)) label = tf.zeros((1, )) train_dataset = tf.data.Dataset.from_tensors( (input_image, label)).batch(batch_size) checkpoint_pr = dcgan.get_checkpoint_prefix() dcgan_obj = dcgan.Dcgan(epochs, enable_function, batch_size) dcgan_obj.train(train_dataset, checkpoint_pr)
def test_one_epoch_without_function(self): epochs = 1 batch_size = 1 enable_function = False input_image = tf.random.uniform((28, 28, 1)) label = tf.zeros((1,)) train_dataset = tf.data.Dataset.from_tensors( (input_image, label)).batch(batch_size) checkpoint_pr = dcgan.get_checkpoint_prefix() dcgan_obj = dcgan.Dcgan(epochs, enable_function, batch_size) dcgan_obj.train(train_dataset, checkpoint_pr)