Closed coder-zc closed 1 year ago
Hi~thank you for your interest in our work. Yes, now we only support GPU machines, we will support CPU machines in the future.
And you can try to modify pykt/datasets/data_loader.py
where we use from torch.cuda import FloatTensor, LongTensor
to set the default machine to GPU, maybe you can change this line to support CPU.
Thank you for your reply!i tried to change from torch.cuda import FloatTensor, LongTensor
into from torch import FloatTensor, LongTensor
, which can run in the CPU.
Thank you for your results, we will update this to a future version.
Hello! i want to konw why generate window_sequences when testing. Could you briefly describe the sequence_window of how and why to do this,which generate a large number of sequences.
Hello! i want to konw why generate window_sequences when testing. Could you briefly describe the sequence_window of how and why to do this,which generate a large number of sequences.
Hello ! here has some discussions, if that does not answer your question, please let us know.
Excuse me, can the model run only in the GPU environment that supports cuda?