News

Model summary in PyTorch with code

model.summary in keras gives a very fine visualization of your model and it's very convenient when it comes to debugging the network..The aim is to provide information complementary to, what is not provided by print(your_model) in PyTorch.
Want to leave a comment?