can your model / bifpn be saved in tensorboard?

[XueZ-phd]

To visualize your network architecture more clearly, I intend to save the network architecture including model and bifpn into a tensorboard.

I add the following codes into your classification/main.py:

writer = SummaryWriter('./tensorboard/'+args_path)
images, labels = next(iter(train_loader))
images = images.cuda()
writer.add_graph(model, images)
writer.add_graph(bifpn, images)

However, it returns many erros.

Therefore, I want to find some help from you to visualize your network architecture using a tensorboard.