search

JoyHuYY1412 / DDE_CIL

58 stars 7 forks source link

error : AttributeError: module 'torch' has no attribute 'flatten' #6

Closed miss-rain closed 3 years ago

miss-rain commented 3 years ago

my pytorch vision is 0.4.0, when i run run.sh, i meet error : AttributeError: module 'torch' has no attribute 'flatten'

Could you help me?

JoyHuYY1412 commented 3 years ago

AttributeError: module 'torch' has no attribute 'flatten'

Hi, don't worry about it. You can use .view() to reshape the tensor or other similar ways.