diff --git a/utilz.py b/utilz.py index d5b4a7f..7a5c54a 100644 --- a/utilz.py +++ b/utilz.py @@ -56,7 +56,7 @@ def wrapper(*args, **kwargs): from tqdm import tqdm as _tqdm def tqdm(a): - return _tqdm(a) if CONFIG().tqdm else a + return _tqdm(a) if config.CONFIG.tqdm else a def squeeze(lol): @@ -151,7 +151,7 @@ def LongVar(array, requires_grad=False): def Var(array, requires_grad=False): ret = Variable(torch.Tensor(array), requires_grad=requires_grad) - if CONFIG.cuda: + if config.CONFIG.cuda: ret = ret.cuda() return ret @@ -167,25 +167,31 @@ def init_hidden(batch_size, cell): hidden = Variable(torch.zeros(layers, batch_size, cell.hidden_size)) context = Variable(torch.zeros(layers, batch_size, cell.hidden_size)) - if CONFIG.cuda: + if config.CONFIG.cuda: hidden = hidden.cuda() context = context.cuda() return hidden, context if isinstance(cell, (nn.GRU, nn.GRUCell)): hidden = Variable(torch.zeros(layers, batch_size, cell.hidden_size)) - if CONFIG.cuda: + if config.CONFIG.cuda: hidden = hidden.cuda() return hidden class Averager(list): - def __init__(self, filename=None, *args, **kwargs): + def __init__(self, filename=None, ylim=None, *args, **kwargs): super(Averager, self).__init__(*args, **kwargs) + self.filename = filename + self.ylim = ylim if filename: - open(filename, 'w').close() + try: + f = '{}.pkl'.format(filename) + if os.path.isfile(f): + log.debug('loading {}'.format(f)) + self.extend(pickle.load(open(f, 'rb'))) + except: + open(filename, 'w').close() - self.filename = filename - @property def avg(self): if len(self):