Model Visualization

[alessa (Aless Bandrabur)]

You can check here Visualizing your network in PyTorch

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