41ef0bf4ece3daf13a02b14c79eb7ddebb42bc0c,facegen/generate.py,,generate_from_yaml,#,418

Before Change

    print("Generating images...")

    gen = model.predict(inputs, batch_size=batch_size)

    print("Writing to "{}"...".format(output_dir))

    for i in range(0, gen.shape[0]):

After Change

    for idx in tqdm(range(0, num_images, batch_size)):

        batch = {
            "identity":    inputs["identity"]   [idx:idx+batch_size,:],
            "emotion":     inputs["emotion"]    [idx:idx+batch_size,:],
            "orientation": inputs["orientation"][idx:idx+batch_size,:],
        }

Instances