def __init__(self, model, gpu=-1): NNBase.__init__(self, model, gpu) self.optimizer = Opt.Adam() self.optimizer.setup(self.model) self.loss_function = F.mean_squared_error self.loss_param = {}
def __init__(self, model, gpu=-1): NNBase.__init__(self, model, gpu) self.optimizer = Opt.Adam() self.optimizer.setup(self.model) self.loss_function = F.softmax_cross_entropy self.loss_param = {}