| |
|
|
| """ |
| Misc download and visualization helper functions and class wrappers. |
| """ |
|
|
| import sys |
| import time |
| import torch |
| from visdom import Visdom |
|
|
|
|
| def reporthook(count, block_size, total_size): |
| global start_time |
| if count == 0: |
| start_time = time.time() |
| return |
| duration = time.time() - start_time |
| progress_size = int(count * block_size) |
| speed = int(progress_size / (1024 * duration)) |
| percent = min(int(count * block_size * 100 / total_size), 100) |
| sys.stdout.write("\r...%d%%, %d MB, %d KB/s, %d seconds passed" % |
| (percent, progress_size / (1024 * 1024), speed, duration)) |
| sys.stdout.flush() |
|
|
|
|
| class VisdomWrapper(Visdom): |
| def __init__(self, *args, env=None, **kwargs): |
| Visdom.__init__(self, *args, **kwargs) |
| self.env = env |
| self.plots = {} |
|
|
| def init_line_plot(self, name, |
| X=torch.zeros((1,)).cpu(), |
| Y=torch.zeros((1,)).cpu(), **opts): |
| self.plots[name] = self.line(X=X, Y=Y, env=self.env, opts=opts) |
|
|
| def plot_line(self, name, **kwargs): |
| self.line(win=self.plots[name], env=self.env, **kwargs) |
|
|
