From 41d2d460d094ae87a88648d18bf283d7b3abc367 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Sun, 2 May 2021 14:51:55 -0700 Subject: [PATCH] link to yannic --- README.md | 2 ++ 1 file changed, 2 insertions(+) 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