Commit 6310782e authored by xhlulu's avatar xhlulu

Add note about pretrained shorthands

parent 75191acc
...@@ -23,6 +23,14 @@ model = torch.hub.load( ...@@ -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: 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