(self)
| 71 | |
| 72 | @unittest.skipIf(torch.cuda.device_count() <= 1, 'distributed unittest') |
| 73 | def test_predict_dist(self): |
| 74 | model_type = 'YoloXPredictor' |
| 75 | model_path = PRETRAINED_MODEL_YOLOXS_EXPORT |
| 76 | self._base_predict(model_type, model_path, dist=True) |
| 77 | |
| 78 | def test_predict_oss_path(self): |
| 79 | model_type = 'YoloXPredictor' |
nothing calls this directly
no test coverage detected