search

rapidsai / crossfit

Metric calculation library
Apache License 2.0
2 stars 6 forks source link

[BUG] Fix logging of prediction pipelines #55

Closed VibhuJawa closed 4 weeks ago

VibhuJawa commented 5 months ago

We should fix logging of prediction pipelines, where we show each GPU as GPU-0. 

GPU: 0, Part: 53: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9493/9493 [03:11<00:00, 49.50it/s]
GPU: 0, Part: 7: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9679/9679 [03:14<00:00, 49.74it/s]
GPU: 0, Part: 61: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9669/9669 [03:13<00:00, 49.85it/s]
GPU: 0, Part: 62: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9656/9656 [03:14<00:00, 49.59it/s]
GPU: 0, Part: 6: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9619/9619 [03:16<00:00, 49.05it/s]
GPU: 0, Part: 63: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9652/9652 [03:15<00:00, 49.36it/s]
GPU: 0, Part: 57: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9676/9676 [03:16<00:00, 49.12it/s]
GPU: 0, Part: 59: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9606/9606 [03:15<00:00, 49.23it/s]
GPU: 0, Part: 56: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9675/9675 [03:18<00:00, 48.73it/s]
GPU: 0, Part: 60: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9595/9595 [03:18<00:00, 48.23it/s]
GPU: 0, Part: 58: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9652/9652 [03:18<00:00, 48.55it/s]
GPU: 0, Part: 8: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9473/9473 [03:16<00:00, 48.09it/s]
GPU: 0, Part: 55: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9871/9871 [03:22<00:00, 48.80it/s]
GPU: 0, Part: 9: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9662/9662 [03:21<00:00, 47.89it/s]