Skip to content

Commit 122e114

Browse files
committed
added link to pretrained models
1 parent e86f440 commit 122e114

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tiatoolbox/models/engine/engine_abc.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,8 @@ def _initialize_model_ioconfig(
264264

265265
if isinstance(model, str):
266266
# ioconfig is retrieved from the pretrained model in the toolbox.
267+
# list of pretrained models in the TIA Toolbox is available here:
268+
# https://tia-toolbox.readthedocs.io/en/add-bokeh-app/pretrained.html
267269
# no need to provide ioconfig in EngineABC.run() this case.
268270
return get_pretrained_model(model, weights)
269271

0 commit comments

Comments
 (0)