From 0af7402daa94660d3b8002f37be6a5fcf9cce9f4 Mon Sep 17 00:00:00 2001
From: chenfeiyu <chenfeiyu@baidu.com>
Date: Thu, 18 Feb 2021 19:33:41 +0800
Subject: [PATCH] add rank_zero_only for ExperimentBase.save

---
 parakeet/training/experiment.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/parakeet/training/experiment.py b/parakeet/training/experiment.py
index 482e012..94caa66 100644
--- a/parakeet/training/experiment.py
+++ b/parakeet/training/experiment.py
@@ -124,6 +124,7 @@ class ExperimentBase(object):
         """
         dist.init_parallel_env()
 
+    @mp_tools.rank_zero_only
     def save(self):
         """Save checkpoint (model parameters and optimizer states).
         """