diff options
| author | zhang <zch921005@126.com> | 2022-07-09 10:17:30 +0800 |
|---|---|---|
| committer | zhang <zch921005@126.com> | 2022-07-09 10:17:30 +0800 |
| commit | bc28bcb9c373574c02ad14bb3c208bae5575412b (patch) | |
| tree | a6ce88e0eb4f9f15545868cd0536b4e5049a6d79 | |
| parent | b116d9f9a5e0aa2bcf48d86fdd9bba92ecf1b526 (diff) | |
update
| -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 |