diff options
| -rw-r--r-- | fine_tune/bert/tutorials/03_bert_input_embedding.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/fine_tune/bert/tutorials/03_bert_input_embedding.py b/fine_tune/bert/tutorials/03_bert_input_embedding.py index 38c56cb..7a58db3 100644 --- a/fine_tune/bert/tutorials/03_bert_input_embedding.py +++ b/fine_tune/bert/tutorials/03_bert_input_embedding.py @@ -19,3 +19,4 @@ model.eval() with torch.no_grad(): output = model(**model_input) +nn.MultiheadAttention
\ No newline at end of file |