espnet2.asr.state_spaces.s4.rank_zero_only

source

espnet2.asr.state_spaces.s4.rank_zero_only(fn: Callable) → Callable

Decorator function from PyTorch Lightning.

Function that can be used as a decorator to enable a function/method being called only on global rank 0.