Есть ли альтернатива tpu_trainer_lib?

Question

Я пытался запустить Google Tensor2tensor ( https://github.com/tensorflow/tensor2tensor) локально, но он задыхается, когда доходит до этого бита:

from tensor2tensor.tpu import tpu_trainer_lib

Что имеет смысл, потому что в моем Mac нет TPU. Есть ли альтернатива этой функции?

Вот сообщение об ошибке

ModuleNotFoundError: No module named 'tensor2tensor.tpu'

Answer 1

Вы, вероятно, сможете заменить его на:

из tenor2tensor.utils импорт trainer_lib

И затем вызвать соответствующие методы из trainer_lib.

Answer 2

Кажется, вы смешали разные версии Tensor2Tensor. В последней версии нет tpu_trainer_lib (он был переименован). Даже в тех немногих версиях, где такой файл существовал, он поддерживал обучение на TPU, GPU и CPU, есть только несколько ветвей if-then-else для кода, специфичного для TPU. Кстати, в настоящее время TPU доступны только для сотрудников Google.