I don't think training on CPU would be a good idea, but I need to make tracking from a trained agent possible on CPU. This would require removing some asserts that force GPU usage, as well as handling the device passed out the code in a much cleaner way.
AntoineTheb
commented
1 year ago
I don't think training on CPU would be a good idea, but I need to make tracking from a trained agent possible on CPU. This would require removing some
asserts that force GPU usage, as well as handling the device passed out the code in a much cleaner way.