From 11ec74905aa5722e50dc62b53800e9cdf4886acc Mon Sep 17 00:00:00 2001
From: max yue <yx931112@qq.com>
Date: Wed, 20 Jan 2021 00:39:49 +0800
Subject: [PATCH] Update integrations.py (#9652)

File "/share/apps/anaconda3/envs/my_env/lib/python3.7/site-packages/transformers/integrations.py", line 419, in __init__
    self._SummaryWriter = SummaryWriter
UnboundLocalError: local variable 'SummaryWriter' referenced before assignment
---
 src/transformers/integrations.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py
index f1519d8c0a8..8692eefdc36 100644
--- a/src/transformers/integrations.py
+++ b/src/transformers/integrations.py
@@ -415,8 +415,9 @@ class TensorBoardCallback(TrainerCallback):
                     self._SummaryWriter = SummaryWriter
                 except ImportError:
                     self._SummaryWriter = None
+        else:
+            self._SummaryWriter = None
         self.tb_writer = tb_writer
-        self._SummaryWriter = SummaryWriter
 
     def _init_summary_writer(self, args, log_dir=None):
         log_dir = log_dir or args.logging_dir