Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
import valohai
def main(old_config):
pipeline = valohai.Pipeline(name="train-superbai", config=old_config)
# Define nodes
convert = papi.execution("convert-superbai")
weights = papi.execution("weights")
train = papi.execution("train")
detect = papi.execution("detect")
# Configure pipeline
convert.output("classes.txt").to(train.input("classes"))
convert.output("train/*").to(train.input("train"))
convert.output("test/*").to(train.input("test"))
convert.output("classes.txt").to(detect.input("classes"))
weights.output("model/*").to(train.input("model"))
train.output("model/*").to(detect.input("model"))
return pipeline
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment