def __enter__(self):
chk = ModelCheckpoint(self.checkpoint, verbose=0, save_best_only=False,
save_weights_only=False, mode='auto')
csv_logger = CSVLogger('training.log')
snaps = LambdaCallback(on_epoch_end=lambda epoch, logs: self.snap(epoch))
return [chk, csv_logger, snaps]
评论列表
文章目录