Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-08 19:58:10 -08:00
parent 1bf717ef9c
commit 37fa9afaff
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
utils/torch_utils.py
View File

@ -101,6 +101,7 @@ def load_classifier(name='resnet101', n=2):
from collections import defaultdict from collections import defaultdict
from torch.optim import Optimizer from torch.optim import Optimizer
class Lookahead(Optimizer): class Lookahead(Optimizer):
def __init__(self, optimizer, k=5, alpha=0.5): def __init__(self, optimizer, k=5, alpha=0.5):
self.optimizer = optimizer self.optimizer = optimizer