diff --git a/setup.py b/setup.py index 1e70b41..51f3735 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ #!/usr/bin/env python from setuptools import find_packages, setup -version = '0.7.3' +version = '0.7.4' setup( name='test_tube', diff --git a/test_tube/log.py b/test_tube/log.py index c01015a..80589b9 100644 --- a/test_tube/log.py +++ b/test_tube/log.py @@ -153,6 +153,13 @@ def __init__( # set the tensorboardx log path to the /tf folder in the exp folder log_dir = self.get_tensorboardx_path(self.name, self.version) + # this is a fix for pytorch 1.1 since it does not have this attribute + for attr, val in [('purge_step', None), + ('max_queue', 10), + ('flush_secs', 120), + ('filename_suffix', '')]: + if not hasattr(self, attr): + setattr(self, attr, val) super().__init__(log_dir=log_dir, *args, **kwargs) # register on exit fx so we always close the writer