开源识别图片深度学习简单框架

易语言 2020-10-13 11:46:24

深度学习简单框架
功能: 卷积2d层
池化2d层(最大池化)
全连接层(线性)
relu(线性整流)
LogSoftmax
MSELoss
nlllloss(一般和LogSoftmax结合使用的)

其他:两个运算方式,别想了没有GPU,都是CPU,一个是语法实现(类_运算),一个是汇编实现(类_汇编),GPU没有能力弄也不想弄。。

如何更换运算方式: 类_模型->私有成员->运算 修改变量的数据类型就行了。