未验证 提交 53da62b0 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #1395 from littletomatodonkey/dev/add_refer

add reference
......@@ -17,6 +17,11 @@ import numpy as np
class ExponentialMovingAverage():
"""
Exponential Moving Average
Code was heavily based on https://github.com/Wanger-SJTU/SegToolbox.Pytorch/blob/master/lib/utils/ema.py
"""
def __init__(self, model, decay, thres_steps=True):
self._model = model
self._decay = decay
......
......@@ -18,6 +18,7 @@ __all__ = ['AverageMeter']
class AverageMeter(object):
"""
Computes and stores the average and current value
Code was based on https://github.com/pytorch/examples/blob/master/imagenet/main.py
"""
def __init__(self, name='', fmt='f', postfix="", need_avg=True):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册