File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ You can see the full script in
4747- ``args `` and ``kwargs `` are passed along to the real callable function.
4848- Docstring of the function works as a help message. It explains what does the model do and what
4949 are the allowed positional/keyword arguments. It's highly recommended to add a few examples here.
50- - Entrypoint function should ** ALWAYS ** return a model(nn.module).
50+ - Entrypoint function can either return a model(nn.module), or auxiliary tools to make the user workflow smoother, e.g. tokenizers .
5151- Pretrained weights can either be stored locally in the github repo, or loadable by
5252 ``torch.hub.load_state_dict_from_url() ``. In the example above ``torchvision.models.resnet.resnet18 ``
5353 handles ``pretrained ``, alternatively you can put the following logic in the entrypoint definition.
You can’t perform that action at this time.
0 commit comments