Add initial model interface (#30)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Ayush Chaurasia
2022-10-25 21:51:15 +02:00
committed by GitHub
co-authored by pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
parent 7b560f7861
commit 1054819a59
12 changed files with 220 additions and 109 deletions
+13
View File
@@ -0,0 +1,13 @@
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()