Using Tensorboard (TypeError: can't pickle _thread.lock objects)

[mahayat]

Hi!
I am trying to incorporate tensorboard with the following snippet in the train function
if args.gpu == 0:
args.tb.add_scalar('loss/train', loss.item(), (len(train_loader)*epoch)+i)
args.tb.add_scalar('acc1/train', acc1[0], (len(train_loader)*epoch)+i)

But I am receiving TypeError: can't pickle _thread.lock objects error originating from mp.spawn().

Any way out?

[mahayat]

FIXED : My Bad

The SummaryWriter() should be in the main_worker function. It was in main