From ab801aaf79e856fecb8fad8fedc2cb17ce768f32 Mon Sep 17 00:00:00 2001 From: Ananya Harsh Jha Date: Tue, 13 Jul 2021 05:12:48 -0400 Subject: [PATCH] updated link to cifar-10 weights for simclr (#686) Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> --- docs/source/self_supervised_models.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/self_supervised_models.rst b/docs/source/self_supervised_models.rst index 715380ef8c..e14f347e6d 100644 --- a/docs/source/self_supervised_models.rst +++ b/docs/source/self_supervised_models.rst @@ -324,7 +324,7 @@ CIFAR-10 pretrained model:: from pl_bolts.models.self_supervised import SimCLR - weight_path = 'https://pl-bolts-weights.s3.us-east-2.amazonaws.com/simclr/bolts_simclr_imagenet/simclr_imagenet.ckpt' + weight_path = 'https://pl-bolts-weights.s3.us-east-2.amazonaws.com/simclr/simclr-cifar10-sgd/simclr-cifar10-sgd.ckpt' simclr = SimCLR.load_from_checkpoint(weight_path, strict=False) simclr.freeze()