Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix pytorch writer #65

Merged
merged 8 commits into from
Nov 30, 2019
Merged

fix pytorch writer #65

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
6956052
fix missing self.purge_step in pt-1.1
Borda Nov 29, 2019
30fbe8c
add comment
Borda Nov 29, 2019
5ddcc4e
version
Borda Nov 29, 2019
4120ea6
fix max_queue
Borda Nov 29, 2019
7cbbf0b
fix flush_secs
Borda Nov 29, 2019
d63af34
fix filename_suffix
Borda Nov 29, 2019
e4cd6df
fix filename_suffix
Borda Nov 29, 2019
94156a0
fix default
Borda Nov 30, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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',
Expand Down
7 changes: 7 additions & 0 deletions test_tube/log.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down