mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-24 06:14:55 +08:00
1054819a59
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
14 lines
332 B
Python
14 lines
332 B
Python
from ultralytics.yolo import YOLO
|
|
|
|
|
|
def test_model():
|
|
model = YOLO()
|
|
model.new("assets/dummy_model.yaml")
|
|
model.model = "squeezenet1_0" # temp solution before get_model is implemented
|
|
# model.load("yolov5n.pt")
|
|
model.train(data="imagenette160", epochs=1, lr0=0.01)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_model()
|