ChainerのTrainer ExtensionでEpochを取得したい

やりたいこと

以下のように, Epoch毎に呼び出されるChainerのTrainer Extensionで, その時のEpochを取得したい.

@training.make_extension(trigger=(10, 'epoch'))
def print_epoch(trainer):
    print(f"Epoch : ")

trainer.extend(print_epoch)

解決

trainer.updater.epochを参照する.

@training.make_extension(trigger=(10, 'epoch'))
def print_epoch(trainer):
    print(f"Epoch : {trainer.updater.epoch}")

trainer.extend(print_epoch)