Hi @HarryZhang67, which documentation are you following? Seems like it might be showing outdated information. The recommended way to report metrics is to use ray.train.report:
ray.train.report(dict(loss=avg_loss, KL_loss=avg_KL_loss, SSIM_loss=avg_SSIM_loss))