提交 84559698 编写于 作者: J jrzaurin

fixed naming of the output file

上级 91e2e48a
......@@ -115,13 +115,13 @@ class History(Callback):
def on_train_begin(self, logs:Optional[Dict]=None):
self.epoch = []
self.history = {}
self._history = {}
def on_epoch_end(self, epoch:int, logs:Optional[Dict]=None):
logs = logs or {}
self.epoch.append(epoch)
for k, v in logs.items():
self.history.setdefault(k, []).append(v)
self._history.setdefault(k, []).append(v)
class ModelCheckpoint(Callback):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册