#tensorflow #stylegan
#tensorflow #stylegan
Вопрос:
Возможно ли запустить версию stylegan2 с TensorFlow на процессоре? Если, то какие изменения необходимо внести для этого?
Комментарии:
1. Я тоже ищу то же самое, кто-нибудь знает, как запустить его на CPU? @Thunder
2. У меня пока нет источника. Основная часть, в которой stylegan2 использует графический процессор, находится во встроенных ядрах CUDA. Но я проверил их операционные файлы ( github.com/NVlabs/stylegan2/tree/master/dnnlib/tflib/ops ), и есть реализация ‘ref’, которая позволяет запускать его на процессоре. После установки impl = ‘ref’ вместо impl = ‘cuda’ везде и установки используемого устройства = CPU везде я застрял с ошибкой (точно не помню), которая была связана с форматом данных (NCHW, NHWC)