torch.no_grad vs. requires_grad

Chen0136 · Jul 3, 2022 · 3e8c688 · 3e8c688
1 parent 8131406
commit 3e8c688
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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.eval()
 with torch.no_grad():
     output = model(**model_input)
+