tracklab
tracklab copied to clipboard
Is there a way to set mps as false while running in Mac
Hello,
I am trying to run https://github.com/SoccerNet/sn-gamestate?tab=readme-ov-file, getting an exception while running https://github.com/SoccerNet/sn-gamestate/issues/34
Is there a way that in tracklab, I set the torch initialisation to explicitly use cpu instead of mps?
Hello,
The device assignment is handled here. You can modify the behavior directly in this function if you need a different setup.
Hope this helps! Let me know if you have further questions.