diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-11-04 02:27:18 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-11-04 02:27:18 -0700 |
| commit | 6b8e96c4bbd73a1e1d4e126d795a26fd0dae983c (patch) | |
| tree | 67072a0442b705b5d5b29840f4b41e13af1d4597 /util/image_pool.py | |
| parent | 5f858eb70a3c110238f74a592bad0e7be601c539 (diff) | |
add update_html_freq flag
Diffstat (limited to 'util/image_pool.py')
| -rw-r--r-- | util/image_pool.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/util/image_pool.py b/util/image_pool.py index 9f34a09..5a242e6 100644 --- a/util/image_pool.py +++ b/util/image_pool.py @@ -2,6 +2,8 @@ import random import numpy as np import torch from torch.autograd import Variable + + class ImagePool(): def __init__(self, pool_size): self.pool_size = pool_size |