Skip to content

Commit b56f30a

Browse files
committed
transfer learning example
1 parent b5eddb3 commit b56f30a

File tree

3 files changed

+654
-5
lines changed

3 files changed

+654
-5
lines changed

README.md

+5-4
Original file line numberDiff line numberDiff line change
@@ -220,10 +220,6 @@ A collection of various deep learning architectures, models, and tips for Tensor
220220
   [PyTorch: [GitHub](pytorch_ipynb/ordinal/ordinal-cnn-beckham2016-afadlite.ipynb) | [Nbviewer](https://nbviewer.jupyter.org/github/rasbt/deeplearning-models/blob/master/pytorch_ipynb/ordinal/ordinal-cnn-beckham2016-afadlite.ipynb)]
221221

222222

223-
224-
225-
226-
227223
## Tips and Tricks
228224

229225
- Cyclical Learning Rate
@@ -234,6 +230,11 @@ A collection of various deep learning architectures, models, and tips for Tensor
234230
   [PyTorch: [GitHub](pytorch_ipynb/tricks/gradclipping_mlp.ipynb) | [Nbviewer](https://nbviewer.jupyter.org/github/rasbt/deeplearning-models/blob/master/pytorch_ipynb/tricks/gradclipping_mlp.ipynb)]
235231

236232

233+
## Transfer Learning
234+
235+
- Transfer Learning Example (VGG16 pre-trained on ImageNet for Cifar-10)
236+
237+
   [PyTorch: [GitHub](pytorch_ipynb/transfer/transferlearning-vgg16-cifar10-1.ipynb) | [Nbviewer](https://nbviewer.jupyter.org/github/rasbt/deeplearning-models/blob/master/pytorch_ipynb/transfer/transferlearning-vgg16-cifar10-1.ipynb)]
237238

238239
## PyTorch Workflows and Mechanics
239240

pytorch_ipynb/cnn/cnn-alexnet-cifar10.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565
"id": "rH4XmErYj5wm"
6666
},
6767
"source": [
68-
"# Model Zoo -- AlexNet CIFAR-10 Classifier"
68+
"# AlexNet CIFAR-10 Classifier"
6969
]
7070
},
7171
{

0 commit comments

Comments
 (0)