Skip to content

moves tensor to CPU before calling .numpy()

fbf7e6f
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Merged

adds available_device to test_confusion_matrix.py #3335 #3357

moves tensor to CPU before calling .numpy()
fbf7e6f
Select commit
Loading
Failed to load commit list.