Delen via

PyTorchConfiguration Klas

Vertegenwoordigt configuratie-informatie voor gedistribueerde PyTorch-taken.

Deze klasse wordt gebruikt in de RunConfiguration klasse.

Klasse PyTorchConfiguration-constructor.

Constructor

PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)

Parameters

Name Description
communication_backend
str

Communicatieback-end die moet worden gebruikt voor gedistribueerde pyTorch-training. Ondersteunde back-ends zijn Nccl en Gloo. De standaardwaarde is 'Nccl'.

Default value: Nccl
process_count
int

Het totale aantal processen dat moet worden gestart voor de taak. Standaard wordt de waarde ingesteld op de node_count.

Default value: None
node_count
int

Het aantal knooppunten dat moet worden gebruikt voor de taak.

Default value: 1