Open minestoo opened 10 months ago
The code is almost identical to the RCD except the contrasive learning, and the error occurred in this module. Did you really complete the experiment?
Our code runs in the provided environment on the NVIDIA 3090 GPU, and dimension errors are usually caused by incorrect environments or devices.
I have adjusted the version of python package, but the tensor dimension is still incorrect.