Hey @ptrblck ! (Apologies for @)
I was just wondering, How can you customize the Tensor-board’s Summary Writer that we have in PyTorch  to something as shown in the below picture?

image587×566 108 KB

Thanks for your response,
Aditya.