Enrollments Open for Certification Course on Artificial Intelligence and Deep Learning by IIT Roorkee

  Apply Now

Project - Training from Scratch vs Transfer Learning

Project - Training from Scratch vs Transfer Learning

0% completed
1 Concept | 7 Assessments | 86 Learners

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