Access Intermediate Layers

[nestarz]

I am using TensorFlow.js and I am interested in gettings the embeddings instead of the logits of a pre-trained model just as described in https://github.com/tensorflow/tfjs-models/tree/master/mobilenet for MobileNet ('Getting embeddings')

With the classic MobileNet model loaded const model = await mobilenet.load() you can use model.infer(image, 'conv_preds') for example and it works fine and you can do your transfer learning.

Any advice would be appreciated !