Skip to content

Commit

Permalink
Update fine-tuning.md - small change
Browse files Browse the repository at this point in the history
Fixed a reference to an image
  • Loading branch information
benliu0001 authored Mar 18, 2024
1 parent 553eb2a commit ceb9d56
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Topics/Tech_Stacks/fine-tuning-ml/fine-tuning.md
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ There is a sidebar containing code that can be useful for learning how others ha
## Setting up Training

Before training, it is important to note that if a GPU is available, using it would speed up training time significantly.
![GPU testing](./Screenshot%202024-03-17%20225750.png)
![GPU testing](./DetectingGPU.png)
To send a PyTorch tensor or PyTorch Module to the GPU, use the method ".to(device)".

![Freezing and Overwriting](./OverWritingLastWeights.png)
Expand All @@ -92,4 +92,4 @@ In this document, we have explored the process of fine-tuning a pre-trained mach

We delved into the specifics of setting up the training process, including freezing the weights of the pre-trained model to preserve its feature extraction capabilities, and replacing the final layer of the model to adapt it to the new task.

Having learned what model fine-tuning is and its applications, I hope any reader of this can add this useful skill to their arsenal of software engineering/machine learning skills.
Having learned what model fine-tuning is and its applications, I hope any reader of this can add this useful skill to their arsenal of software engineering/machine learning skills.

0 comments on commit ceb9d56

Please sign in to comment.