Skip to content

Commit

Permalink
Add note about pretrained shorthands
Browse files Browse the repository at this point in the history
  • Loading branch information
xhlulu committed Nov 6, 2021
1 parent 75191ac commit 6310782
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 @@ -23,6 +23,14 @@ model = torch.hub.load(
)
```

Currently, the following `pretrained` shorthands are available:
```python
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="celeba_distill")
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="face_paint_512_v1")
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="face_paint_512_v2")
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="paprika")
```

You can also load the `face2paint` util function. First, install dependencies:

```
Expand Down

0 comments on commit 6310782

Please sign in to comment.