Solved by ptrblck in post #2 
              
              
                print(model) will print the registered modules, which are not present in your current implementation. 
You could create the extra_repr method, which would print additional information about your custom module, similar to this one. 
To print the parameters, you could use: 
for name, param in model.na…