search

roeehendel / icl_task_vectors

82 stars 21 forks source link

Tensors on different device while running the `exploration.ipynb` notebook #2

Open sanchit-ahuja opened 9 months ago

sanchit-ahuja commented 9 months ago

Hi,

Thank you for your excellent work. I have been trying to mess around with your code by running the exploration.ipynb file. However, I am stuck with this error of tensors on being on different device. I am using a machine with 8 GPUs. The main experiments script works out-of-the-box but it does not use any GPU (Getting 0 GPU utilization). Can you help me with this? image

gowtham1997 commented 7 months ago

did you manage to fix this?

sanchit-ahuja commented 7 months ago

nope