I'm working on a machine without a GPU, and it would be really nice to be able to use STAR without cuda for developing. From my understanding, most of the code imports torch.cuda so when GPU isn't available, it throws an error. Would it be possible to disable that?
I'm working on a machine without a GPU, and it would be really nice to be able to use STAR without cuda for developing. From my understanding, most of the code imports
torch.cuda
so when GPU isn't available, it throws an error. Would it be possible to disable that?