TPU дает сбой на том же шаге во время обучения с ошибкой Tensor имел значения NaN. Тот же код работал вчера, но без каких-либо изменений он не работал на другом устройстве TPU с количеством ядер. Кто-нибудь может подсказать, как это исправить?