logger.info("[{}/{}][{}/{}] => Loss D = {} -- Loss G = {} (time spent: {})".format(epoch,n_epochs,i,len(dataloader),errD.data[0],errG.data[0],(end-start)))
# save generated images at every epoch
# TODO: model moved to CPU and back and I don't really know why (expected CPU tensor error)
# To summarize:
# tried to move tensors, variables on the GPU -> does not work