赞
踩
1、定义网络(加载网络)
2、加载数据与真值label/GT(dataset、dataloader)
3、将数据投喂到网络中,计算预测值pred
4、计算loss(pred与GT的差异)
5、反向传播BP,更新参数
6、迭代3-5步骤
如上述的3、查看engine/model步骤
将步骤2加载的数据投喂到网络中,计算预测值pred;将pred与GT比较,在前向传播中计算loss
核心模块
model
engine
nn
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。