Skip to content

Fix parameter count in ModelSummary when parameters are DTensors #21404

Fix parameter count in ModelSummary when parameters are DTensors

Fix parameter count in ModelSummary when parameters are DTensors #21404

Triggered via pull request August 4, 2024 23:34
@awaelchliawaelchli
synchronize #20163
Status Skipped
Total duration 2s
Artifacts

tpu-tests.yml

on: pull_request_target
Matrix: test-on-tpus
Fit to window
Zoom out
Zoom in