Can pytorch-accelerated be used in other types of tasks, such as detection, reconstruction? #21
-
pytorch-accelerated is great. However, how can we adapt it to tasks other than classification? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 1 reply
-
Hi @cekxm, thanks for the kind words! PyTorch-accelerated can be used for lots of tasks other than classification, for example, this blog post demonstrates using it for a recommendation problem and I have recently been using it to train Yolox for object detection (blog post hopefully coming soon)! In many cases, you can get away with using the trainer directly as long as your dataset returns (xb, yb) that are passed to a loss function. In cases where that doesn't work, you will have to subclass the trainer, such as this example. Were there any tasks that you had in mind? |
Beta Was this translation helpful? Give feedback.
Hi @cekxm, thanks for the kind words! PyTorch-accelerated can be used for lots of tasks other than classification, for example, this blog post demonstrates using it for a recommendation problem and I have recently been using it to train Yolox for object detection (blog post hopefully coming soon)!
In many cases, you can get away with using the trainer directly as long as your dataset returns (xb, yb) that are passed to a loss function. In cases where that doesn't work, you will have to subclass the trainer, such as this example.
Were there any tasks that you had in mind?