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 411
411
" # https://github.com/bioimage-io/core-bioimage-io-python/blob/main/bioimageio/core/build_spec/build_model.py#L252\n " ,
412
412
" cite = [{\" text\" : cite_entry.text, \" url\" : cite_entry.url} for cite_entry in model_resource.cite]\n " ,
413
413
" \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 " ,
414
417
" # the axes descriptions for the inputs / outputs\n " ,
415
418
" input_axes = [\" bcyx\" ]\n " ,
416
419
" output_axes = [\" bcyx\" ]\n " ,
441
444
" architecture=model_source,\n " ,
442
445
" model_kwargs=model_resource.weights[\" pytorch_state_dict\" ].kwargs,\n " ,
443
446
" preprocessing=preprocessing,\n " ,
444
- " postprocessing=postprocessing\n " ,
447
+ " postprocessing=postprocessing,\n " ,
448
+ " training_data=training_data,\n " ,
445
449
" )"
446
450
]
447
451
},
You can’t perform that action at this time.
0 commit comments