Skip to content

Commit b92e19e

Browse files
Add training data to example notebook
1 parent 186cb1e commit b92e19e

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

example/bioimageio-core-usage.ipynb

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -411,6 +411,9 @@
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",
@@ -441,7 +444,8 @@
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
},

0 commit comments

Comments
 (0)