pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard
verified publisher icon Cadene

wrong with senet???

Open sxzy opened this issue 7 years ago • 2 comments

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

I don't know what's wrong here.

sxzy avatar Dec 06 '18 05:12 sxzy

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

Cadene avatar Dec 13 '18 20:12 Cadene

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

sxzy avatar Dec 14 '18 04:12 sxzy