def __init__(self, symbol, ctx=None,
begin_epoch=0, num_epoch=None,
arg_params=None, aux_params=None,
optimizer='sgd', **kwargs):
# ??????
self.symbol = symbol
if ctx is None:
ctx = mx.cpu()
self.ctx = ctx
# ??epoch,??0?????????
self.begin_epoch = begin_epoch
# ???epoch?
self.num_epoch = num_epoch
self.arg_params = arg_params
self.aux_params = aux_params
# ??????,?????????
self.optimizer = optimizer
self.kwargs = kwargs.copy()
评论列表
文章目录