83e80f82d7cf5aa0444c9bdc8709304f10b8d7fc,docs/_static/examples/gan.py,,saver_callback,#Any#,115

Before Change

def saver_callback(state):
    batches_done = state[tb.EPOCH] * len(state[tb.GENERATOR]) + state[tb.BATCH]
    if batches_done % sample_interval == 0:
        save_image(state[GEN_IMGS].data[:25], "images/%d.png" % batches_done, nrow=5, normalize=True)


// Configure data loader

After Change

def saver_callback(state):
    batches_done = state[tb.EPOCH] * len(state[tb.GENERATOR]) + state[tb.BATCH]
    if batches_done % sample_interval == 0:
        samples = state[tb.MODEL].generator(batch)
        save_image(samples, "images/%d.png" % batches_done, nrow=5, normalize=True)

Instances