Module_Combine.py 文件源码

def __init__(self, name, symbol, 
               data_names=('data',), 
               data_shapes=(),
               label_names=('label'), 
               label_shapes=(),
               inputs_need_grad=False,
               optimizer='sgd',
               optimizer_params={'learning_rate':0.1, 'momentum':0.9, 'wd':0.0005},
               initializer=mx.init.Normal(),
               context=mx.cpu()):
    self.name = name
    self.symbol = symbol
    self.data_names = data_names
    self.label_names = label_names
    self.data_shapes = data_shapes
    self.label_shapes = label_shapes
    self.inputs_need_grad = inputs_need_grad
    self.optimizer = optimizer
    self.optimizer_params = optimizer_params
    self.initializer = initializer
    self.context = context