Spaces:
Sleeping
Sleeping
Update safety_analyzer/train_yolo.py
Browse files
safety_analyzer/train_yolo.py
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
```python
|
| 2 |
from ultralytics import YOLO
|
| 3 |
import logging
|
| 4 |
import torch
|
|
@@ -47,4 +46,3 @@ def train_model():
|
|
| 47 |
if __name__ == "__main__":
|
| 48 |
logger.info("Starting YOLOv8 model training...")
|
| 49 |
train_model()
|
| 50 |
-
```
|
|
|
|
|
|
|
| 1 |
from ultralytics import YOLO
|
| 2 |
import logging
|
| 3 |
import torch
|
|
|
|
| 46 |
if __name__ == "__main__":
|
| 47 |
logger.info("Starting YOLOv8 model training...")
|
| 48 |
train_model()
|
|
|