Pipe Pretrained Model to Custom Layers

[ptrblck]

You could use forward hook as described here and register the hook on the fc6 layer.