Fine-Tuning Machine Learning Models with DVC Experiments for Transfer Learning by@FlippedCoding
529 reads
529 reads

Fine-Tuning Machine Learning Models with DVC Experiments for Transfer Learning

by Milecia22mSeptember 3rd, 2021
Read on Terminal Reader
Read this story w/o Javascript
tldt arrow

Too Long; Didn't Read

There are plenty of machine learning models available that have been trained to solve one problem and the knowledge gained from that can be applied to a new, yet related problem. For example, a model like AlexNet has been trained on millions of images so you could potentially use this to classify cars, animals, or even people. In this post, we'll be fine-tuning the AlexNet model and the SqueezeNet model to classify bees and ants. We start by initializing these models so we can get the number of model features and the input size we need. We'll use DVC to handle experiments for us and we'll compare the results.

Companies Mentioned

Mention Thumbnail
Mention Thumbnail
featured image - Fine-Tuning Machine Learning Models with DVC Experiments for Transfer Learning
Milecia HackerNoon profile picture
Milecia

Milecia

@FlippedCoding

Software/Hardware Engineer | International tech speaker | Random inventor and slightly mad scientist

Learn More
LEARN MORE ABOUT @FLIPPEDCODING'S
EXPERTISE AND PLACE ON THE INTERNET.

Share Your Thoughts

About Author

Milecia HackerNoon profile picture
Milecia@FlippedCoding
Software/Hardware Engineer | International tech speaker | Random inventor and slightly mad scientist

TOPICS

THIS ARTICLE WAS FEATURED IN...

Permanent on Arweave
Read on Terminal Reader
Read this story in a terminal
 Terminal
Read this story w/o Javascript
Read this story w/o Javascript
 Lite
L O A D I N G
. . . comments & more!