You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I‘m a rookie, can anyone give me some advice? Thanks a lot!
The environment is:
Python 3.7
CUDA 10.0
Pytorch 1.1.0
Torchvision 0.3.0
By the way, I use Google Colab to do this job. The default version of CUDA is 11.2, but the version I need is 10.0. I noticed that there is CUDA 10.0 under path “usr/loacl” , I pointed the soft connection of "usr/local/CUDA" to "usr/local/CUDA10.0", but When I check the GPU information, it shows CUDA 11.2 which confused me.
The text was updated successfully, but these errors were encountered:
Sagiri18
changed the title
An error occured when I train the model on my own dataset.
An error occured when I trained the model on my own dataset.
Mar 17, 2021
Hey buddy, check detclasslist.txt to see if it starts from 0. I solved the issue by correcting this part.
i meet same problem by start at 1,should it start at 0 or 1?If start at 0 i assmue backgroud is 0?If start at 1 does that means when set num_classes we should add 1?(if we got 8 classes we should write 9 for config instead)
When I recurrent the R-C3D model to achieve the action recognization task on my own dataset, I had the same problem.
I‘m a rookie, can anyone give me some advice? Thanks a lot!
The environment is:
By the way, I use Google Colab to do this job. The default version of CUDA is 11.2, but the version I need is 10.0. I noticed that there is CUDA 10.0 under path “usr/loacl” , I pointed the soft connection of "usr/local/CUDA" to "usr/local/CUDA10.0", but When I check the GPU information, it shows CUDA 11.2 which confused me.
The text was updated successfully, but these errors were encountered: