Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Updates
Browse files Browse the repository at this point in the history
  • Loading branch information
ethanwharris committed May 19, 2021
1 parent 4fd0344 commit c21e816
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion docs/source/template/backbones.rst
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,8 @@ You can create a registry like this:
TEMPLATE_BACKBONES = FlashRegistry("backbones")
Let's add a simple MLP backbone to our registry.
We'll create the backbone and return it along with the output size (so that we can create the model head in our :class:`~flash.core.model.Task`).
We need a function that creates the backbone and returns it along with the output size (so that we can create the model head in our :class:`~flash.core.model.Task`).
You can use any name for the function, although we use ``load_{model name}`` by convention.
Here's the code:

.. literalinclude:: ../../../flash/template/classification/backbones.py
Expand Down

0 comments on commit c21e816

Please sign in to comment.