I am trying to train the MS2FP model exactly based on the demo colab and input data from repo. Training works with cpu. Also prediction works with GPU. However when I try training with GPU I get the following error:
Pointer argument (at 2) cannot be accessed from Triton (cpu tensor?)
It runs without error until the following line in this file: IDSL_MINT-main/IDSL_MINT/MS2FP/MS2FP_Model.py:
Source = self.PeakEmbedding(MZ_Tokens, INT).permute(1, 0, 2)
Do you have any idea what can be the problem?
Thanks!
I am trying to train the MS2FP model exactly based on the demo colab and input data from repo. Training works with cpu. Also prediction works with GPU. However when I try training with GPU I get the following error:
Pointer argument (at 2) cannot be accessed from Triton (cpu tensor?)It runs without error until the following line in this file: IDSL_MINT-main/IDSL_MINT/MS2FP/MS2FP_Model.py:
Source = self.PeakEmbedding(MZ_Tokens, INT).permute(1, 0, 2)Do you have any idea what can be the problem?
Thanks!