FullyShardedDataParallel

FullyShardedDataParallel

Определение

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

Где встречается

Навигация