Skip to content

Latest commit

 

History

History
78 lines (54 loc) · 1.89 KB

index.rst

File metadata and controls

78 lines (54 loc) · 1.89 KB

chainermn

API Reference

Communicators

create_communicator

CommunicatorBase

Optimizers and Evaluators

create_multi_node_optimizer

create_multi_node_evaluator

chainermn.extensions.GenericMultiNodeEvaluator

Dataset Utilities

scatter_dataset

scatter_index

chainermn.datasets.create_empty_dataset

MultiNodeChainList

chainermn.links.MultiNodeBatchNormalization

chainermn.links.create_mnbn_model

Functions

chainermn.functions.send

chainermn.functions.recv

chainermn.functions.pseudo_connect

chainermn.functions.bcast

chainermn.functions.gather

chainermn.functions.scatter

chainermn.functions.alltoall

chainermn.functions.allgather

Iterators

chainermn.iterators.create_multi_node_iterator

chainermn.iterators.create_synchronized_iterator

Trainer extensions

chainermn.extensions.AllreducePersistent

chainermn.extensions.multi_node_snapshot

chainermn.create_multi_node_checkpointer

Configurations

configurations