9923b06 e4da1db b165569
1
2
3
4
5
6
7
8
9
import os def process_rank() -> int: return int(os.environ.get("RANK", os.environ.get("SLURM_PROCID", 0))) is_rank_zero = process_rank() == 0
