RankZeroCallback

class composer.core.callback.RankZeroCallback

Bases: composer.core.callback.Callback, abc.ABC

Base class for callbacks that only run on the rank zero process.

Note

init() and load_state_dict() are executed before the DDP fork and will be called on all ranks.