mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 11:06:35 +00:00
add export to FP16
add inference with possibility to have different
This commit is contained in:
@@ -184,6 +184,7 @@ def train_dataset(existing_date=None, from_scratch=False):
|
||||
model_name = latest_model if latest_model is not None and path.isfile(latest_model) and not from_scratch else 'yolo11m.yaml'
|
||||
print(f'Initial model: {model_name}')
|
||||
model = YOLO(model_name)
|
||||
model.info['author'] = 'LLC Azaion'
|
||||
|
||||
yaml = abspath(path.join(cur_dataset, 'data.yaml'))
|
||||
results = model.train(data=yaml,
|
||||
|
||||
Reference in New Issue
Block a user