To practice convolutional neural network (CNN) design, I worked on a classification problem that was more challenging than toy datasets but still lightweight enough to train quickly on my local machine. The following details the design of a 3-block CNN architecture and the lessons I learned from training it on clothing image data. Dataset Characteristics…
Tag: PyTorch
PyTorch is a flexible, open-source deep learning framework designed to simplify building, training, and deploying neural networks for computer vision.