About using tensorboardX to visualize the network structure

facebookresearch / SparseConvNet

Submanifold sparse convolutional networks

https://github.com/facebookresearch/SparseConvNet

Other

2.05k stars 333 forks source link

About using tensorboardX to visualize the network structure #100

Closed fruityangyang closed 5 years ago

fruityangyang commented 5 years ago

Thank you for sharing your amazing code! Now, i'm trying to use tensorboardX to visualize the network structure, but it failed. Can tensorboardX be used to visualize the sparseconvnet? If don't, which tool do you recommend to visualize the network structure?

btgraham commented 5 years ago

scn.Sequential inherits from torch.nn.Sequential so standard methods might work. I have not really looked into this yet, I am afraid. Ben