plt torch support, test=model

Zeref996 · Jan 2, 2025 · 3faaca1 · 3faaca1
1 parent 8229719
commit 3faaca1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/framework/e2e/PaddleLT_new/engine/torch_train.py b/framework/e2e/PaddleLT_new/engine/torch_train.py
@@ -128,7 +128,6 @@ def torch_dy_train(self):
             opt = optimizer.get_opt(net=net)
 
         for epoch in range(self.step):
-            print("data is: ", data)
             logit = net(*data)
             # 构建loss用于训练
             dy_loss = loss.get_loss(logit)