ML ZOOMCAMP 2025 - Module 8
Module 8: Neural networks and Deep learning
In module we learned how to build an image classification model using PyTorch and transfer learning. We used a clothing dataset.
What was covered
- Introduction to PyTorch for deep learning
- Loading and preprocessing image data
- Using pre-trained models (MobileNetV2)
- Understanding convolutional neural networks (CNNs)
- Transfer learning: adapting pre-trained models
- Hyperparameter tuning: learning rate optimization
- Model checkpointing: saving the best model
- Adding more layers to improve performance
- Dropout regularization to prevent overfitting
- Data augmentation for better generalization
- Training the final model
- Using the model for predictions
- Exporting models to ONNX format
Comments
Post a Comment