wrong with senet??? - Githubissues

Cadene / pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

BSD 3-Clause "New" or "Revised" License

9.01k stars 1.83k forks source link

wrong with senet??? #110

Open sxzy opened 5 years ago

sxzy commented 5 years ago

Hello, I have tried to load the senet ,and use the pretrained one ....but the error as below

I don't know what's wrong here.

Cadene commented 5 years ago

It is due to pytorch2.7. Unfortunately pretrainedmodels is not totally compatible for now.

sxzy commented 5 years ago

It is due to pytorch2.7. Unfortunately pretrainedmodels is not totally compatible for now.

I change the code in definition of SEResNeXtBottleneck(Bottleneck), width = math.floor(planes (base_width / 64)) groups ====> width = int(math.floor(planes (base_width / 64)) groups)

then it is ok