CIFAR10-VGG19-Tensorflow
CIFAR10-VGG19-Tensorflow copied to clipboard
Transfer learning with tensorflow
Hi,
I am using yolo architecture and trying transfer learning. I have pre-trained weights trained over 80 classes but now I want output for 3 classes only. How can I remove the last convolution layer and fit my own layer?
Below is the initializer method where I am restoring model and training only the last part but I want to change last convolutional layer as well.
Thanks