Skip to content

Update batch_eth_mnist.py#667

Merged
Hananel-Hazan merged 1 commit intoBindsNET:masterfrom
AXYZdong:master
Mar 21, 2024
Merged

Update batch_eth_mnist.py#667
Hananel-Hazan merged 1 commit intoBindsNET:masterfrom
AXYZdong:master

Conversation

@AXYZdong
Copy link
Contributor

@AXYZdong AXYZdong commented Jan 7, 2024

add .to(device) in line 209, 214, 366, 368.
If not, may have the error RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

@Hananel-Hazan Hananel-Hazan merged commit fb32876 into BindsNET:master Mar 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants