David Heidelberg 13659ac6fa examples: numpy() array returns only one value, not an array (#1534) ...

Fixes issue:
```
    loss_cpu = loss.detach().numpy()[0]
               ~~~~~~~~~~~~~~~~~~~~~^^^
IndexError: too many indices for array: array is 0-dimensional, but 1 were indexed
```

Signed-off-by: David Heidelberg <david@ixit.cz>

2023-08-13 14:33:05 -07:00

2.2 KiB

Executable File

Raw Blame History

View Raw