esm icon indicating copy to clipboard operation
esm copied to clipboard

Published 20 hours ago • verified publisher icon facebookresearch

Allow inverse folding model to score sequences on GPU

Open hnisonoff opened this issue 2 years ago • 0 comments

I modified the get_sequence_loss function to use whichever device the model is currently on to also be the device that the batch of sequences are put on.

hnisonoff avatar Oct 27 '22 05:10 hnisonoff