File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 411411 " # https://github.com/bioimage-io/core-bioimage-io-python/blob/main/bioimageio/core/build_spec/build_model.py#L252\n " ,
412412 " cite = [{\" text\" : cite_entry.text, \" url\" : cite_entry.url} for cite_entry in model_resource.cite]\n " ,
413413 " \n " ,
414+ " # the training data used for the model can also be specified by linking to a dataset available on bioimage.io\n " ,
415+ " training_data = {\" id\" : \" ilastik/stradist_dsb_training_data\" }\n " ,
416+ " \n " ,
414417 " # the axes descriptions for the inputs / outputs\n " ,
415418 " input_axes = [\" bcyx\" ]\n " ,
416419 " output_axes = [\" bcyx\" ]\n " ,
441444 " architecture=model_source,\n " ,
442445 " model_kwargs=model_resource.weights[\" pytorch_state_dict\" ].kwargs,\n " ,
443446 " preprocessing=preprocessing,\n " ,
444- " postprocessing=postprocessing\n " ,
447+ " postprocessing=postprocessing,\n " ,
448+ " training_data=training_data,\n " ,
445449 " )"
446450 ]
447451 },
You can’t perform that action at this time.
0 commit comments