Model Visualization

You can check here Visualizing your network in PyTorch

For printing the content of your model just type learn.model

2 Likes