Skip to content

Conversation

aaprasad
Copy link

Add patch for batch model with (1,N, B) shape eg when iterating thru a torch dataloader. Also added a .to method to make it easy to move the model to and from gpu

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.

1 participant