hezhangsprinter / DCPDN

Densely Connected Pyramid Dehazing Network (CVPR'2018)
401 stars 112 forks source link

about pytorch's version #8

Closed KosumosuL closed 6 years ago

KosumosuL commented 6 years ago

after changing the version of pytorch to 0.3.1 the following problem occurred:

Traceback (most recent call last):
  File "demo.py", line 128, in <module>
    netG = net.dehaze(inputChannelSize, outputChannelSize, ngf)
  File "/data2/zhangzhengxi4047/dcpdn.pytorch/DCPDN/dehaze22.py", line 540, in __init__
    self.tran_dense=Dense()
  File "/data2/zhangzhengxi4047/dcpdn.pytorch/DCPDN/dehaze22.py", line 411, in __init__
    haze_class = models.densenet121(pretrained=True)
  File "build/bdist.linux-x86_64/egg/torchvision/models/densenet.py", line 27, in densenet121
  File "build/bdist.linux-x86_64/egg/torchvision/models/densenet.py", line 213, in __init__
AttributeError: 'module' object has no attribute 'kaiming_normal_'

what should i do?

KosumosuL commented 6 years ago

solved

Neobe commented 6 years ago

Hello,I meet the same problem,can you tell me how to solve this?

KosumosuL commented 6 years ago

Sorry for my forgetting the specific solution, but it since like I solved this problem by re-installing the pytorch from github.

2018-07-26 14:40 GMT+08:00 Neobe notifications@github.com:

Hello,I meet the same problem,can you tell me how to solve this?

— You are receiving this because you modified the open/close state. Reply to this email directly, view it on GitHub https://github.com/hezhangsprinter/DCPDN/issues/8#issuecomment-407993165, or mute the thread https://github.com/notifications/unsubscribe-auth/AeKVrO_LKk96O7uSBYmluUAVK5V2rFEtks5uKWR2gaJpZM4UdzAO .