YashashGaurav

# 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