New ViT backbone, validation, and code optimizations#40

Merged

loribonna merged 11 commits intomasterfrom dev

Jun 16, 2024

+3,385-2,596

Commits on Jun 2, 2024

Commits on Jun 14, 2024

Add cifar10-224. Add system_tracker to log memory usage. Add code_optimization argument. Minor format changes.
loribonna
committed

Commits on Jun 16, 2024

Add validation_mode flag. Updates to validation.
loribonna
committed
Updated validation for domain/general cl.
loribonna
committed
Add validation and fitting modes. Add early stopping. Updated docs. Add tests for checkpoint. Minor refactors.
loribonna
committed