Tensorflow 2.0：示例不会训练

Question

我正在尝试运行DCGAN，以及本教程中的其他示例： https://www.tensorflow.org/beta/tutorials/generative/dcgan

该培训不存在或非常缓慢。好像卡住了。

我已经修改了train函数。输出为(256,28,28,1)（输出形状），然后依次为a和b。直到我停止它为止，这种情况已经持续了很长时间。没有完成一个时代。 （打印速度很慢）

请参见下面的修改版本：

def train(dataset, epochs):
    for epoch in range(epochs):
        start = time.time()

        for image_batch in dataset: #stuck in an infinite loop
            print('a')
            train_step(image_batch)
            print('b')

        print('c')
        # Produce images for the GIF as we go
        display.clear_output(wait=True)
        generate_and_save_images(generator,epoch + 1,seed)
        print('d')

        # Save the model every 1 epochs
        if (epoch + 1) % 1 == 0:
            checkpoint.save(file_prefix = checkpoint_prefix)

        print ('Time for epoch {} is {} sec'.format(epoch + 1, time.time()-start))

此外，我正在使用tensorflow-gpu==2.0.0-beta1，并使用NVIDIA GTX 960M。

旁注：在Google Collab上，它显然可以工作，但我希望它可以在笔记本电脑上工作。

也在CycleGAN上，是培训的屏幕截图：