search

open-mmlab / mmengine

OpenMMLab Foundational Library for Training Deep Learning Models
https://mmengine.readthedocs.io/
Apache License 2.0
1.11k stars 331 forks source link

Integration torchinfo #708

Open vansin opened 1 year ago

vansin commented 1 year ago

bdab56019de5cdf38af366a483e3b32

xin-li-67 commented 1 year ago

It will be great to have this feature in mmengine~~

RangeKing commented 1 year ago

Hope to integrate torchinfo

C1rN09 commented 1 year ago

Seems cool! We'll discuss on it