Is there any function for summarizing model?

Questions
#1

Hi everyone,
Is there any function to print the summary of a model in PyTorch/Keras?
Like the model.summary() method does in Keras. Or the torchinfo summary method in Pytorch?

Thank you all.

#2

Currently DGL does not support torchinfo because torchinfo assumes that the arguments are all PyTorch tensors, which is not the case for DGL models where you also pass in DGL graph objects.

#3

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.