Hi Driazati,
Thanks for the codes, it helps. just some typo i think, the
print_modules(module.module, level + 1);
should be
print_modules(module, level + 1);

Appreciate your help. The references for these seems to be not much, could you share some good reference for this ? For e.g., the method (or the var) that can return the input size of a pre-trained network etc.
Thanks again.
Regards,
CL