Add device arg to torch train and predict #188

funkelab / gunpowder

A library to facilitate machine learning on multi-dimensional images.

https://funkelab.github.io/gunpowder/

MIT License

78 stars 56 forks source link

Closed Mohinta2892 closed 11 months ago

Mohinta2892 commented 1 year ago

Pre-Merge Checklist:

[x] if this PR adds a feature: request merge into the latest development branch (patch-1.3.2)
[ ] if this PR fixes a bug: request merge into the latest patch branch (patch-X.Y.Z)
[x] PR branch name is short and describes the feature/bug addressed in this PR #188
[x] commit messages are consistent
[ ] changes reviewed by another contributor
[x] tests cover changes
[x] all tests pass

pattonw commented 1 year ago

please target patch-1.3.2, and run black to make spacings consistent

Mohinta2892 commented 1 year ago

Hi Will,

Thanks, reformatted with black!

pattonw commented 1 year ago

Hi Samia, Thanks! but no need to reformat unrelated files, just the ones you modified would be great

Mohinta2892 commented 1 year ago

Sorry, fixed.

pattonw commented 11 months ago

squashed and merged into the patch branch