Pytorch xla против Pytorch Ligthning: мульти-TPU

Я пытаюсь использовать функцию мульти-TPU Colab. Мне было интересно, что лучше использовать, Pytorch xla или Pytorch Lightning.

Источник

1 ответ

Решение

Pytorch Lightning построен с использованием xla; таким образом, синтаксис xla больше похож на pytorchs, а на lightning - нет.

Источник
Другие вопросы по тегам