Skip to content
This repository was archived by the owner on Mar 21, 2024. It is now read-only.

SimCLR recovery test#643

Merged
ant0nsc merged 25 commits into
mainfrom
maxilse/lr_scheduler_recovery_test
Jan 24, 2022
Merged

SimCLR recovery test#643
ant0nsc merged 25 commits into
mainfrom
maxilse/lr_scheduler_recovery_test

Conversation

@maxilse

@maxilse maxilse commented Jan 24, 2022

Copy link
Copy Markdown
Contributor

The test checks if a SSLContainer correctly resumes training.
First we run SSL using a Trainer for 20 epochs.
Second, we run a new SSL job for 15 epochs.
Third we resume the job and run it for 5 more epochs.
The test checks the learning rate and the loss.
The test needs a GPU to run successfully.

In addition to the test itself, the PR contains a dummy data class and a dummy SSLContainer that could be useful for other tests.

@maxilse maxilse requested a review from ant0nsc January 24, 2022 16:51
@ant0nsc ant0nsc merged commit 6a4919c into main Jan 24, 2022
@ant0nsc ant0nsc deleted the maxilse/lr_scheduler_recovery_test branch January 24, 2022 18:56
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants