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

Conversation

@maxilse
Copy link
Contributor

@maxilse maxilse commented Jan 24, 2022

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.

4 participants