深度学习简单框架
功能: 卷积2d层
池化2d层(最大池化)
全连接层(线性)
relu(线性整流)
LogSoftmax
MSELoss
nlllloss(一般和LogSoftmax结合使用的)
其他:两个运算方式,别想了没有GPU,都是CPU,一个是语法实现(类_运算),一个是汇编实现(类_汇编),GPU没有能力弄也不想弄。。
如何更换运算方式: 类_模型->私有成员->运算 修改变量的数据类型就行了。
下一篇 随机自动生成软件注册码
深度学习简单框架
功能: 卷积2d层
池化2d层(最大池化)
全连接层(线性)
relu(线性整流)
LogSoftmax
MSELoss
nlllloss(一般和LogSoftmax结合使用的)
其他:两个运算方式,别想了没有GPU,都是CPU,一个是语法实现(类_运算),一个是汇编实现(类_汇编),GPU没有能力弄也不想弄。。
如何更换运算方式: 类_模型->私有成员->运算 修改变量的数据类型就行了。