Last active
May 17, 2022 04:26
-
-
Save minyoung/485b24a8ec827d59dfe2f6bf5c47d59e to your computer and use it in GitHub Desktop.
truncated tensorboard event loading
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
import logging | |
import threading | |
import time | |
logging.basicConfig() | |
logging.getLogger("tensorboard").setLevel(logging.DEBUG) | |
from tensorboard.backend.event_processing import directory_watcher | |
from tensorboard.backend.event_processing import event_file_loader | |
generator = directory_watcher.DirectoryWatcher(".", event_file_loader.EventFileLoader) | |
finish = threading.Event() | |
def iterate_tb(): | |
time.sleep(0.1) | |
while True: | |
for event in generator.Load(): | |
print("a", event.step) | |
if finish.is_set(): | |
break | |
print("a", "done") | |
t = threading.Thread(target=iterate_tb) | |
t.start() | |
finish.set() | |
for event in generator.Load(): | |
print("b", event.step) | |
print("b", "done") | |
t.join() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment