AttributeError: 'int' object has no attribute 'lower'
出现问题原因:
self.conv1 = Conv2D(filters=6, kernel_size=5, padding='same')
当使用Conv2d层搭建网络,需要使用padding进行填充,但是Tensorflow的padding参数值只有两种可选方式,分别是same
和valid
,前者是保持输入和输出图像尺寸不变,它会自动计算应该填充多少,默认是valid
会造成图像尺寸减小,pytorch支持使用数字进行填充。