From dd2960d8d57e3558aa43157d309fb2d1f97ba0a0 Mon Sep 17 00:00:00 2001 From: Eggry Ran Date: Fri, 13 Jan 2023 07:21:12 +0800 Subject: [PATCH 1/4] update SwAV example --- docs/source/models/self_supervised.rst | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/docs/source/models/self_supervised.rst b/docs/source/models/self_supervised.rst index 27054b271d..b5493a6a56 100644 --- a/docs/source/models/self_supervised.rst +++ b/docs/source/models/self_supervised.rst @@ -489,12 +489,13 @@ To Train:: gpus=1, num_samples=dm.num_unlabeled_samples, dataset='stl10', - batch_size=batch_size + batch_size=batch_size, + nmb_crops=(2,4) ) # fit - trainer = pl.Trainer(precision=16) - trainer.fit(model) + trainer = pl.Trainer(precision=16, accelerator='gpu', devices=1) + trainer.fit(model, datamodule=dm) Pre-trained ImageNet ******************** From bd23c27f3adc7712a268c22e19edf05b78f7b117 Mon Sep 17 00:00:00 2001 From: Jirka B Date: Fri, 19 May 2023 10:17:31 -0400 Subject: [PATCH 2/4] update mergify team --- .github/mergify.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/mergify.yml b/.github/mergify.yml index dc431ff4c5..314ae28dca 100644 --- a/.github/mergify.yml +++ b/.github/mergify.yml @@ -79,4 +79,4 @@ pull_request_rules: actions: request_reviews: teams: - - "@PyTorchLightning/core-bolts" + - "@Lightning-Universe/core-Bolts" From afdc09aa7761cf1dc4b960701d780f1972be75b3 Mon Sep 17 00:00:00 2001 From: Jirka B Date: Fri, 19 May 2023 10:17:31 -0400 Subject: [PATCH 3/4] update mergify team --- .github/mergify.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/mergify.yml b/.github/mergify.yml index dc431ff4c5..e53eba78f4 100644 --- a/.github/mergify.yml +++ b/.github/mergify.yml @@ -79,4 +79,4 @@ pull_request_rules: actions: request_reviews: teams: - - "@PyTorchLightning/core-bolts" + - "@Lightning-Universe/core-bolts" From f624a87fe1678823ddc76f5d92100c54a83627f4 Mon Sep 17 00:00:00 2001 From: Eggry Ran Date: Sat, 20 May 2023 00:36:46 +0800 Subject: [PATCH 4/4] Update docs/source/models/self_supervised.rst Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> --- docs/source/models/self_supervised.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/models/self_supervised.rst b/docs/source/models/self_supervised.rst index b5493a6a56..3ede334ec6 100644 --- a/docs/source/models/self_supervised.rst +++ b/docs/source/models/self_supervised.rst @@ -494,7 +494,7 @@ To Train:: ) # fit - trainer = pl.Trainer(precision=16, accelerator='gpu', devices=1) + trainer = pl.Trainer(precision=16, accelerator='auto') trainer.fit(model, datamodule=dm) Pre-trained ImageNet