Commit 6310782e authored by xhlulu's avatar xhlulu

Add note about pretrained shorthands

parent 75191acc
......@@ -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:
```
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment