Python get_checkpoint_prefix Beispiele

Beispiel #1

    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)

Beispiel #2

Datei: dcgan_test.py Projekt: indranig/examples

  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)