BonanDing
/

DeMemWM

DeMemWM / utils /distributed_utils.py

Keep reproduction train and eval fixes

9923b06 about 1 month ago

62 Bytes

	import os

	is_rank_zero = int(os.environ.get("RANK", 0)) == 0