Closed EdenBelouadah closed 2 years ago
I figured out that the problem was coming from the cuda version. I fixed it with
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
Just a remark, the train0 folder shouldn't be empty. Otherwise the program crashes.
Thank you again
Hello, I'm trying to run your pre-trained model on my test dataset, to see the effect on it. This is the content of the subfolders:
The program just doesn't pass this step:
I also notices cuda warning. Could you please help? Thank you