Update 1-1,结构化数据建模流程范例.md

val/test阶段，添加with torch.no_grad()，关闭梯度计算。
Daniel123jia · Jul 5, 2020 · 073f9a7 · 073f9a7
1 parent 027aab7
commit 073f9a7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/1-1,结构化数据建模流程范例.md b/1-1,结构化数据建模流程范例.md
@@ -340,9 +340,11 @@ for epoch in range(1,epochs+1):
     val_step = 1
 
     for val_step, (features,labels) in enumerate(dl_valid, 1):
-
-        predictions = net(features)
-        val_loss = loss_func(predictions,labels)
+        # 关闭梯度计算
+        with torch.no_grad():
+            predictions = net(features)
+            val_loss = loss_func(predictions,labels)
+
         val_metric = metric_func(predictions,labels)
 
         val_loss_sum += val_loss.item()