search

Exscientia / abodybuilder3

Apache License 2.0
40 stars 11 forks source link

Fix nb device error #7

Closed wjs20 closed 4 months ago

wjs20 commented 4 months ago

Not all tensors were loaded onto the GPU so pytorch was complaining. Added everything to gpu if available else cpu. This fixed the issue for me.