Enrollments Open for Certification Course on Artificial Intelligence and Deep Learning by IIT RoorkeeApply Now
Welcome to the project on Training from Scratch vs Transfer Learning. In this exercise, we will understand how to train a neural network from scratch to classify data using TensorFlow 2. We would also learn how to use the weights of an already trained model to achieve classification to another set of data.
We will train a neural network (say model A) on data related to 6 of the classes, and we will train another neural network (say model B) on the remaining 2 classes. Then, we would use the pre-trained weights of model A and tune the last layer so as to classify these 2 classes(this technique is called Transfer Learning), and compare the results of classification obtained using normal training and transfer learning. In this project, we would practically appreciate the use of Transfer Learning.
Machine Learning Engineer @ CloudxLab