Project using PyTorch in which we create custom datasets and dataloaders, train a convnext_tiny model and log it using tensorboard, do inferences and use Captum for more detailed results.
-
Updated
Apr 11, 2025 - Jupyter Notebook