From 37fa9afaff4be124aa573f5d30d316ea783a5aab Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 8 Dec 2019 19:58:10 -0800 Subject: [PATCH] updates --- utils/torch_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index ecbcd306..d984dfca 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -101,6 +101,7 @@ def load_classifier(name='resnet101', n=2): from collections import defaultdict from torch.optim import Optimizer + class Lookahead(Optimizer): def __init__(self, optimizer, k=5, alpha=0.5): self.optimizer = optimizer @@ -165,4 +166,4 @@ class Lookahead(Optimizer): def add_param_group(self, param_group): param_group["counter"] = 0 - self.optimizer.add_param_group(param_group) \ No newline at end of file + self.optimizer.add_param_group(param_group)