当前位置:网站首页>【MindSpore】在训练过程中的step代表什么?

【MindSpore】在训练过程中的step代表什么?

2022-08-10 03:28:00 小乐快乐

1、尝试在ubuntu上进行目标检测算法的训练,使用gpu训练时step显示的效果和官方一样,训练一轮直接显示结果。

2、使用CPU训练时每一次step都显示出来了,这里的step是如何计算的?为什么和gpu显示效果不同?

3.开始训练时出现了以下警告,会不会影响训练。

[WARNING] DEBUG(20487,7f2b2333c740,python3.7):2021-10-09-11:00:52.433.387 [mindspore/ccsrc/debug/debugger/debugger.cc:92] Debugger] Not enabling debugger. Debugger does not support CPU.

epoch和step可以这里理解下:

·epoch:把所有训练样本学习一遍为一个epoch;

·step:每一个step表示进行了一次学习、更新一遍参数,每一次参数更新需要batch size个样本进行学习,根据运算结果调整一次参数。

·batch size:一次迭代(step)所使用的样本数量。

假设有n个样本,batch size为m个样本,epoch为1。那么step=(n×1)/m。

原网站

版权声明
本文为[小乐快乐]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_45666880/article/details/126234368