Fix parameter count in ModelSummary when parameters are DTensors#20163
Merged
awaelchli merged 8 commits intomasterfrom bugfix/model-summary-dtensor-shapesAug 5, 2024
+52-13
Commits
Commits on Aug 4, 2024
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed