while using pipeline with TPU, how can we check whether the device pipeline is using is TPU if we set device_map to “auto”?
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Could I use the device map for pipelines parallel training? | 0 | 243 | April 3, 2023 | |
| 🤗Transformer with Trainer API on TPU VMs and TPU Pods | 0 | 410 | December 18, 2023 | |
| Device_map="auto" with error: Expected all tensors to be on the same device | 7 | 6757 | January 5, 2025 | |
| Machine Translation TPU Script | 0 | 268 | August 20, 2020 | |
| Found a BUG and basic docs code fails to run on kaggle tpu | 0 | 351 | September 15, 2023 |