Moothes

@chunweit Sorry, I have never met the first issue. My environment is Pytorch 1.0.0 and torchvision 0.2.1. For the second issue, "total_ops" and "total_params" are appended by the Thop, which...

> @chunweit the PretrainModel file(not the ImageNet one) given I think is wrong: > in **Experiement.py**, if you print out the model parameters: > > ``` > encoder.encoder.conv1.weight > encoder.encoder.bn1.weight...

@BarCodeReader @chunweit I have updated the weights. You can check it again. Sorry for this mistake.

@chunweit You can check lines 4-10 in model.baseline.py file. If you import the official vgg implementation, this weight file can work.

Your pre-trained pkl file seems not suitable for our code. I can provide my pkl file to you, but it may take a couple of days. Or you can try...

Sorry for this issue. I have uploaded my weight files to BaiduYun and the URLs are listed in readme file. Due to the coronavirus, it took me some time to...

I run with pytorch version 0.4 or 1.0 is OK. From the error message, maybe you can try to replace the 'requires_grad' parameter to a torch.device type parameter.

File "/home/user/***/ITSD-pytorch/models/encoder/resnet.py", line 8, i n import torchvision.models as models You can try to remove this line. It seems to have no effect on the code. My torch version 1.0.0,...

I have updated a new version code using official imagenet-pretrained weights. You can try the new one.

Sorry for your email. I didn't get the notification from my app. All the provided models are imagenet-pretrained weights.