SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update integrations.py (#9652)

Browse Source 
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
This commit is contained in:
max yue
2021-01-20 00:39:49 +08:00
committed by GitHub
parent b020a736c3
commit 11ec74905a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/integrations.py
View File

@@ -415,8 +415,9 @@ class TensorBoardCallback(TrainerCallback):
self._SummaryWriter = SummaryWriter self._SummaryWriter = SummaryWriter
except ImportError: except ImportError:
self._SummaryWriter = None self._SummaryWriter = None
else:
self._SummaryWriter = None
self.tb_writer = tb_writer self.tb_writer = tb_writer
self._SummaryWriter = SummaryWriter
def _init_summary_writer(self, args, log_dir=None): def _init_summary_writer(self, args, log_dir=None):
log_dir = log_dir or args.logging_dir log_dir = log_dir or args.logging_dir