Open yuh-13 opened 3 weeks ago
I tried the main_encoder.py and my environment and parameters are all set (including cuda=0), but the code could only run with the cpu not the gpu I wish for. I used GTX3090 for the experiment.
I tried the main_encoder.py and my environment and parameters are all set (including cuda=0), but the code could only run with the cpu not the gpu I wish for. I used GTX3090 for the experiment.