torch # core deep learning library (CPU or GPU) torchvision # image datasets, transforms, models
Pillow # required for image loading (used by torchvision) numpy # numerical arrays and utilities
tqdm # progress bars (optional but useful) matplotlib # optional for visualizing samples, losses, etc.
scikit-learn # optional: metrics like accuracy_score, confusion_matrix
https://docs.google.com/document/d/1-V-il0oCzyS4GQJz2t_1L6HYi0Thl__i1lUdXuCtuTo/edit?usp=sharing