Skip to content

Commit

Permalink
Add transfer learning section to readme.
Browse files Browse the repository at this point in the history
  • Loading branch information
n2cholas committed Mar 23, 2021
1 parent c682ab0 commit 1367f91
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,14 @@ match exactly. Feel free to use it via `pretrained_resnetd` (should be fine for
transfer learning). You must install fast.ai yourself
([instructions](https://docs.fast.ai/)) to use this function.

### Transfer Learning

The `slice_model` function (found in in
[`common.py`](https://github.com/n2cholas/jax-resnet/blob/main/jax_resnet/common.py))
allows you to extract a subsection of the model and variables dict (e.g.
everything but the Dense layer, so you can add your own classification head).
Check out that docstring for more information.

## References

- [Deep Residual Learning for Image Recognition. Kaiming He, Xiangyu Zhang,
Expand Down

0 comments on commit 1367f91

Please sign in to comment.