Skip to content

Commit

Permalink
Update custom_models.md (#1409)
Browse files Browse the repository at this point in the history
In the example of timm models, there is a misuse of MLP( ), which should be model.

Co-authored-by: boyufan24 <buaafby@126.com>
  • Loading branch information
boyufan and boyufan24 authored Jan 30, 2024
1 parent 68b90a1 commit 1a7f3e3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/developer_guides/custom_models.md
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ those are a major building block of this model, we should apply LoRA to the 2D c
those layers, let's look at all the layer names:

```python
print([(n, type(m)) for n, m in MLP().named_modules()])
print([(n, type(m)) for n, m in model.named_modules()])
```

This will print a very long list, we'll only show the first few:
Expand Down

0 comments on commit 1a7f3e3

Please sign in to comment.