From the course: Transfer Learning for Images Using PyTorch: Essential Training

Unlock the full course today

Join today to access over 22,600 courses taught by industry experts or purchase this course individually.

Train the extractor

Train the extractor

From the course: Transfer Learning for Images Using PyTorch: Essential Training

Start my 1-month free trial

Train the extractor

“

- [Instructor] So I'm going to expand the Training the Fixed Feature Extractor section. We're going to be using the Adam optimizer, and the second line sends our model to the GPU. And, finally, we add our model parameters as arguments to the Adam optimizer. Now, the next section of code is Training the Fixed Feature Extractor, which is what we've looked at. So I'm going to run this cell. So this should take between five and 10 minutes. So now that we've completed the section, I'm going to close it off.

Contents