With torch.utils.tensorboard, a model can be viewed in browser via SummaryWriter.add_graph()
Though, I’m unable to view the attributes of an Op
- e.g. kernel size, stride of a Conv2D:
I guess these attributes are passed to Op as normal parameters …?
- Is there anyway to show them in tensorboard?