github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 13:09:56 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Backward compatibility to tensorboard < 2.0.0

Browse source
This commit is contained in:
JanFschr 2021-11-23 10:56:21 +01:00 • committed by GitHub
commit c4a39f57f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
mainscripts/Trainer.py
View file

@ -25,11 +25,15 @@ class TensorBoardTool:
def run(self):
from tensorboard import default
from tensorboard import program
from tensorboard import version as tb_version
# remove http messages
log = logging.getLogger('werkzeug').setLevel(logging.ERROR)
# Start tensorboard server
tb = program.TensorBoard(default.get_plugins())
tb.configure(argv=[None, '--logdir', self.dir_path, '--port', '6006', '--bind_all'])
tb_argv = [None, '--logdir', self.dir_path, '--port', '6006']
if int(tb_version.VERSION[0])>=2:
tb_argv.append("--bind_all")
tb.configure(argv=tb_argv)
url = tb.launch()
print('Launched TensorBoard at {}'.format(url))