Closed XueZ-phd closed 2 years ago
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)
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.
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.