[FIX] Convert bfloat16 to float32 before calling .numpy
#829
| Job | Run time |
|---|---|
| 1m 41s | |
| 1m 41s |
bfloat16 to float32 before calling .numpy
#829
| Job | Run time |
|---|---|
| 1m 41s | |
| 1m 41s |