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

3 years ago · 16d84367c6
parent 089c060756
commit 16d84367c6
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)