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_)
