Merge pull request #248 from EleutherAI/Tee-patch

Update logging.py
EleutherAI · Apr 25, 2021 · 3778af3 · 3778af3
2 parents f90e083 + 4ec6c3f
commit 3778af3
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/megatron/logging.py b/megatron/logging.py
@@ -37,8 +37,17 @@ def __del__(self):
  self.file.close()
 
  def write(self, data):
- self.file.write(data)
- self.std.write(data)
+ try:
+ self.file.write(data)
+ except OSError:
+ pass
+ try:
+ self.std.write(data)
+ except OSError:
+ pass
 
  def flush(self):
- self.file.flush()
+ try:
+ self.file.flush()
+ except OSError:
+ pass