torch.mtia.manual_seed

torch.mtia.manual_seed(seed) [исходный код]

Устанавливает начальное значение для генерации случайных чисел на текущем устройстве MTIA. Эту функцию безопасно вызывать, даже если MTIA недоступна; в таком случае вызов игнорируется без уведомления.

Параметры:

seed (int) – Желаемое начальное значение.

Предупреждение

Если вы работаете с моделью на нескольких GPU, этой функции недостаточно для обеспечения детерминированности. Чтобы задать начальное значение для всех GPU, используйте manual_seed_all().