New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

enhanced w&b logging #269

Merged

MaximilienLC merged 1 commit into main from wandb-mnist

Feb 16, 2024

Collaborator

MaximilienLC commented Feb 16, 2024

Enhanced validation W&B logging (images, y, probs, preds) , see https://wandb.ai/cneuroml/classify_mnist

Changes:

Added logic to log W&B validation data (see BaseLitModule, BaseClassificationLitModule & MNISTClassificationLitModule)
Renamed MNISTClassificationDataModule to MNISTDataModule
Add a seed BaseSubtaskConfig option for reproducibility
No more shuffling val & test dataset (for w&b logging consistency)
Adding cpus_per_task option in MNIST example
Fixed num_workers finder capping at 2 workers


          enhanced w&b logging

c197c94

MaximilienLC merged commit 1c829e7 into main

2 checks passed

MaximilienLC deleted the wandb-mnist branch

February 16, 2024 16:39

MaximilienLC added a commit that referenced this pull request


          enhanced w&b logging (#269)

f5ce032

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet