def AddTrainNet(self, net): Solver.AddTrainNet(self, net) params = self.net_.learnable_params() for i in range(len(params)): blob = Blob() blob.ReshapeLike(params[i]) self.history_.append(blob)
def AddTrainNet(self, net): Solver.AddTrainNet(self, net) params = self.net_.learnable_params() for i in range(len(params)): s = Blob() r = Blob() s.ReshapeLike(params[i]) r.ReshapeLike(params[i]) self.s_.append(s) self.r_.append(r)
def AddTrainNet(self, net): Solver.AddTrainNet(self, net) params = self.net_.learnable_params() for i in range(len(params)): h = Blob() u = Blob() t = Blob() h.ReshapeLike(params[i]) u.ReshapeLike(params[i]) t.ReshapeLike(params[i]) self.history_.append(h) self.update_.append(u) self.temp_.append(t) self.history_.extend(self.history_)