61228f3243eaee39cba31a9abd2afc6ead9612a8,implementations/bicyclegan/bicyclegan.py,,,#,136

Before Change

                    epoch=epoch, batch=i)

    // Update learning rates
    lr_scheduler_G.step()
    lr_scheduler_D_VAE.step()
    lr_scheduler_D_LR.step()

    if opt.checkpoint_interval != -1 and epoch % opt.checkpoint_interval == 0:

After Change

        // Print log
        sys.stdout.write("\r[Epoch %d/%d] [Batch %d/%d] [D VAE_loss: %f, LR_loss: %f] [G loss: %f, pixel: %f, latent: %f] ETA: %s" %
                                                        (epoch, opt.n_epochs,
                                                        i, len(dataloader),
                                                        loss_D_VAE.item(), loss_D_LR.item(),
                                                        loss_GE.item(), loss_pixel.item(),
                                                        loss_latent.item(), time_left))

        if batches_done % opt.sample_interval == 0:
            sample_images(batches_done)

Instances