diff options
| author | zhang <zch921005@126.com> | 2022-07-03 12:55:08 +0800 |
|---|---|---|
| committer | zhang <zch921005@126.com> | 2022-07-03 12:55:08 +0800 |
| commit | 3e8c68838bb39a2a8772ea955ff2f5750375fe1a (patch) | |
| tree | febdadc1be9c033b667d65b2befb8110f1eb9d8a | |
| parent | 8131406d83acad62de49ad51f219b3d2dba562d8 (diff) | |
torch.no_grad vs. requires_grad
| -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 41bdd29..38c56cb 100644 --- a/fine_tune/bert/tutorials/03_bert_input_embedding.py +++ b/fine_tune/bert/tutorials/03_bert_input_embedding.py @@ -18,3 +18,4 @@ model_input = tokenizer(test_sent, return_tensors='pt') model.eval() with torch.no_grad(): output = model(**model_input) + |