Skip to content
Snippets Groups Projects
Commit 89a313e5 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[models.model] Implement to() method to move all computing to the designated... 

[models.model] Implement to() method to move all computing to the designated accelerator device, including losses and augmentations
parent 4a09305d
No related branches found
No related tags found
1 merge request!49Optimise device allocation for all attributes used in the training loop
Pipeline #87958 failed
Stage: qa
Stage: doc
Stage: dist
Stage: test
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment