Manan Goel

@awaelchli This comes from the pytorch-forecasting library which assumes a tensorboard logger and hence uses `add_figure`. The code for that is [here](https://github.com/jdb78/pytorch-forecasting/blob/master/pytorch_forecasting/models/temporal_fusion_transformer/__init__.py#L820).

@LielinJiang @lijianshe02 I was wondering if it was possible to get some feedback

@LielinJiang @LielinJiang I would love some feedback on this whenever you have the time

Hi @LielinJiang @lijianshe02 It would be awesome if I could get this reviewed whenever possible

@qingqing01 @wangxinxin08 Hey guys! I was wondering if I could get some feedback on this PR whenever possible.

Hi @lyuwenyu @ghostxsl I was wondering if it would be possible to get a review for this PR

I was wondering if I could get some feedback on this PR whenever possible

Hi @LDOUBLEV @WenmuZhou I was wondering if it would be possible to get a review on this PR

@WenmuZhou I've made the requested change

@seanlip could I pick this up?