search

ocean-data-factory-sweden / kso

Notebooks to upload/download marine footage, connect to a citizen science project, train machine learning models and publish marine biological observations.
GNU General Public License v3.0
4 stars 12 forks source link

Tutorial 5, issue with model training #340

Closed Bergylta closed 6 months ago

Bergylta commented 6 months ago

image

jannesgg commented 6 months ago

@Bergylta Can you post the code block that is run before?

Bergylta commented 6 months ago 
mlp.train_yolo(
    exp_name=exp_name.value,
    weights=weights.artifact_path,
    project=mlp.project_name,
    epochs=epochs.value,
    batch_size=batch_size.value,
    img_size=img_h.value,  # this requires an int
)