Set workers=0 for MPS Train and Val modes (#4697)

2023-09-02 04:25:03 +02:00 · 2023-09-02 04:25:03 +02:00 · a4fabfdacf
commit a4fabfdacf
parent 2bc6e647c7
14 changed files with 88 additions and 74 deletions
--- a/ultralytics/utils/callbacks/raytune.py
+++ b/ultralytics/utils/callbacks/raytune.py
@ -7,6 +7,7 @@ try:
    import ray
    from ray import tune
    from ray.air import session
+
 except (ImportError, AssertionError):
    tune = None