模型训练与测试-----pytorch和keras的软融合----GAP与全连接的作用
由于模型是由两个混合一起进行软结合,相对于传统上的模型混合都是保存h5,我们这里可以让两个不同的框架在一个框架下软流通。具体如何操作以及注意的点:△ 排布不同首先是pytorch 和keras 的tensor本身的排布是不一样的,这里是需要注意的,避免在后面混合的时候出问题。def ocr_feature(self, x): #input pytorch type: (B,...