SupContrast
SupContrast copied to clipboard
Another bug?
Shall line 126 in main_linear.py
model.load_state_dict(state_dict)
in the same block with return?
For no matter whether cuda is available, we always wants the model to be initiated with ckpt["model"]