issue with cuda

[yuh-13]

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.