how to get the information of input and output nodes

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

3.98k stars 412 forks source link

Open ghost opened 4 years ago

ghost commented 4 years ago

how to get the information of input and output nodes, like this network.

ghost commented 4 years ago