From bc28bcb9c373574c02ad14bb3c208bae5575412b Mon Sep 17 00:00:00 2001 From: zhang Date: Sat, 9 Jul 2022 10:17:30 +0800 Subject: update --- fine_tune/bert/tutorials/03_bert_input_embedding.py | 1 + 1 file changed, 1 insertion(+) (limited to 'fine_tune/bert/tutorials') 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 -- cgit v1.2.3