When doing model prediction, how to get the next-to-last layer emeddings at the same time?

shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.

https://deepctr-torch.readthedocs.io/en/latest/index.html

Apache License 2.0

2.95k stars 696 forks source link

When doing model prediction, how to get the next-to-last layer emeddings at the same time? #271

Closed CFZhai closed 1 year ago

CFZhai commented 1 year ago

I has trained a model using DeepFM. Now I want to get the embedding from the next-to-last layer( or each layer) of test data. How can I make it? Thank you!