def __call__(self, transformer, callback_data, phase, data, idx):
if phase == CallbackPhase.minibatch_post:
if ((idx + 1) % self.frequency == 0):
interval = slice(idx + 1 - self.frequency, idx)
train_cost = callback_data["cost/train"][interval].mean()
tqdm.write("Interval {} Iteration {} complete. Avg Train cost: {}".format(
idx // self.frequency + 1, idx + 1, train_cost))
评论列表
文章目录