EnigmaHong

@shulisiyuan 请问方向识别的数据集是怎么样？是直接将图片分别旋转三个方向吗？

@jiangzz1628 你好，请问找到找到方法解决了吗？我也出现了这个问题。 ValueError: Dimension 1 in both shapes must be equal, but are 5990 and 5531. Shapes are [512,5990] and [512,5531]. for 'Assign_36' (op: 'Assign') with input shapes: [512,5990], [512,5531].

好的，已经跑起来了，感谢

各位，我用1080ti训练，一直提示OUT OF MEMORY，这个训练需要很高的配置吗？你们是用什么训练的？