From 3e8c68838bb39a2a8772ea955ff2f5750375fe1a Mon Sep 17 00:00:00 2001 From: zhang Date: Sun, 3 Jul 2022 12:55:08 +0800 Subject: torch.no_grad vs. requires_grad --- fine_tune/bert/tutorials/03_bert_input_embedding.py | 1 + 1 file changed, 1 insertion(+) (limited to 'fine_tune') 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) + -- cgit v1.2.3