p1atdev/LECO

Issue with training

davizca opened this issue · 1 comments

davizca commented

Hi!

1fbd895b06b2f8f720be7175a48c10be
I'm having this problem trainng LECO... any solution to this?

Thanks in advance

ChrisWang13 commented

Obviously something is wrong with the version of xformers.

You need to verify the version of your cuda and Pytorch to check whether they are compatible with your current xformers.

In setup, you should try:
pip install torch torchvision --index-url https://download.pytorch.org/whl/YOUR_CUDA_VERSION