https://github.com/mert-kurttutan/torchview
maybe can try this one, and can send to logger clearml at the end.