From 6b8e96c4bbd73a1e1d4e126d795a26fd0dae983c Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 4 Nov 2017 02:27:18 -0700 Subject: add update_html_freq flag --- util/image_pool.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'util/image_pool.py') 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 -- cgit v1.2.3-70-g09d2