Ayush Chaurasia 1054819a59
Add initial model interface (#30)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-25 21:51:15 +02:00

5 lines
159 B
Python

from ultralytics.yolo.v8.classify.train import ClassificationTrainer
from ultralytics.yolo.v8.classify.val import ClassificationValidator
__all__ = ["train"]