~BlacKHawK~ YashashGaurav
YashashGaurav
/ log_custom_ml_models.py
Last active
September 19, 2022 22:47
best version of model logging system that I have built for GDrive - Kinda depends on wandb for naming - but easily customizable to any to other service provider.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Model Logging setup | |
| import os | |
| from os.path import isfile, join | |
| def log_checkpoint( | |
| epoch: int, | |
| model: torch.nn.Module, | |
| optimizer: torch.optim.Optimizer, | |
| lr_scheduler: object, # Cuz the base class seemed like a protected instance |