-
What is the proper example of using the output (embedding) of Transformer (or LSTM) as the input feature matrix? Each node in the graph consists of a
|
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 1 reply
-
It depends on whether you want to fine-tune your transformer/LSTM or simply want to use their embeddings (detached from the computation graph). In case of the latter, you can simply input your embeddings into |
Beta Was this translation helpful? Give feedback.
It depends on whether you want to fine-tune your transformer/LSTM or simply want to use their embeddings (detached from the computation graph). In case of the latter, you can simply input your embeddings into
data.x
(as you have done). In case of the former, you would need to input the sequence into the transformer as part of your model. Does that answer your question? :)