Train Models Your Way

from picsellia import Client

client = Client()
project = client.get_project("defects")

# Create experiment
experiment = project.create_experiment("yolo-training")

# Attach dataset
dataset = client.get_dataset("defects").get_version("v3")
experiment.attach_dataset("train", dataset)

from picsellia_cv_engine import step, Pipeline

@step
def train(context):
    model = load_model(context.parameters)
    for epoch in range(context.parameters.epochs):
        # Your training logic
        context.experiment.log("loss", loss)
    context.experiment.store("model.pt")

pipeline = Pipeline([train])
pipeline.run()