fix Tensor.numpy()[0] to float(Tensor) to adapt 0D

3 years ago · 3605078442
parent 6b00ad6064
commit 3605078442
1 changed files with 1 additions and 1 deletions
--- a/examples/tess/cls0/local/train.py
+++ b/examples/tess/cls0/local/train.py
@ -121,7 +121,7 @@ if __name__ == "__main__":
            optimizer.clear_grad()

            # Calculate loss
-            avg_loss += loss.numpy()[0]
+            avg_loss += float(loss)

            # Calculate metrics
            preds = paddle.argmax(logits, axis=1)