search

QiaoranC / tf_ResNeSt_RegNet_model

tensorflow 2.x version of ResNeSt,RegNet,DETR
129 stars 31 forks source link

bug #1

Closed mcmingchang closed 4 years ago

mcmingchang commented 4 years ago

tensorflow-gpu==2.1.0

File "F:\tf_ResNeSt_RegNet_model-master\models\ResNest.py", line 87, in call filters = inputs.shape[self._channel_axis].value AttributeError: 'int' object has no attribute 'value'

mcmingchang commented 4 years ago

I modified it to 'filters = inputs.shape[self._channel_axis]'

QiaoranC commented 4 years ago

Thanks, i test it under tensorflow-gpu==1.15, very similar to 2.X style. I will upgrade to 2.1 and update the code.