diff --git a/README.md b/README.md index 673a8c0..4f5ad93 100644 --- a/README.md +++ b/README.md @@ -426,6 +426,8 @@ torch.save(model.state_dict(), './pretrained-net.pt') You can train `ViT` with the recent SOTA self-supervised learning technique, Dino, with the following code. +Yannic Kilcher video + ```python import torch from vit_pytorch import ViT, Dino