Open Rasoul77 opened 1 year ago
Hi @Rasoul77, by chance did you find a solution?
@Rasoul77 I solved the serialization by creating the ResNet50 model from the old implementation (https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py), changing the first layer kernel size to 7 and deleting the fully connected part and taking only the projection head. The class that represents the network is SupConResNet.
@Rasoul77 Did you find a solution? I got the same error when i ran main_linear.py.
I tried to use the provided pre-trained PTH file presented in the Update section using the following code snippet, but it returns errors of missing keys,
which results,
Could you clarify for which model the pre-trained "supcon.pth" is provided?