Hello,
I have tried to use tensorboard pytorch to plot loss graphs and accuracy.
I have used Summary Writer here.
But, as I run my training loop, I face the following error at the end of first epoch.
Following snippet shows the error:

summarywritererror1148×614 103 KB

Following is the python code containing summary writer for reference:

summarywriter_itr1099×272 20.5 KB


summarywriter_trainer_2966×701 64.4 KB

I also want to plot confusion matrix in tensorboard, please can I have some insights in this as well.
Any help is most appreciated!
Thank you in advance.