Unverified Commit 11ec7490 authored by max yue's avatar max yue Committed by GitHub
Browse files

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
parent b020a736
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment