Google Colab слишком долго обучает GAN

Я следил за учебником по Generative-Adversarial Network на сайте TensorFlow (ссылка ниже), в нем говорится: «Это может занять около одной минуты / эпохи с настройками по умолчанию в Colab.». Пока завершились только две эпохи, каждая из которых составляет около 750 секунд. Это означает, что на прохождение всех 50 эпох потребуется 10 часов. Сначала я попробовал с настройками по умолчанию, затем попробовал с выбранным TPU, и теперь все стало хуже. Я надеюсь найти что-нибудь не так со своей стороны, что могло бы помочь ускорить обучение GAN.

Код слишком длинный, чтобы помещать его здесь, поэтому я помещаю ссылку на свое пошаговое руководство по руководству ниже.

Ссылка на руководство: https://www.tensorflow.org/tutorials/generative/dcgan

Ссылка на мой код Google Colab: https://drive.google.com/open?id=10-VTSyFqWMRT3NIWOU77he7li4YRE0dY




Ответы (1)


Вы пробовали настроить «тип среды выполнения» на графический процессор в меню «Время выполнения»?

person Vidit Gautam    schedule 22.08.2020
comment
Это не дает ответа на вопрос. Как только у вас будет достаточная репутация, вы сможете комментировать любой пост; вместо этого предоставит ответы которые не требуют пояснений от автора вопроса. - Из отзыва - person Eyeslandic; 23.08.2020