当前位置:网站首页>【MindSpore】多卡训练保存权重问题
【MindSpore】多卡训练保存权重问题
2022-07-30 19:04:00 【小乐快乐】
问题描述:
多卡训练的epoch更多吗?单卡训练只需要5个epoch就可以训练好,8卡训练反而需要85epoch?
训练采用数据并行模式。
如下图,我按rank id分开保存的权重是单卡训练的还是多卡合并的结果?/0/crnn-85_4.ckpt和 /1/crnn-85_4.ckpt是同样的吧?

数据并行,每张卡上的ckpt数据是一样的。
/0/crnn-85_4.ckpt和 /1/crnn-85_4.ckpt 应该是一样的。(只有bn的moving_mean和moving_variance可能有差别,但是不影响推理)
单卡训练只需要5个epoch就可以训练好,8卡训练反而需要85epoch?
这个结论是如何来的呢?比较loss值?
每隔一定step 验证一次模型,发现多卡训练反而需要更多epoch才能达到同样精度.单卡可以5epoch达到0.9,多卡需要40-50epoch。学习率,batchsize不变情况下
我的数据集很小,训练集只有5000样本。batchsize是32.验证码集样本500. 学习率是0.0001。是不是数据集小了,切分数据集反而导致每张卡更难收敛,需要的epoch更多?
解决方案:
batch size一致,是总batch一致,还是单卡的bs和数据并行的其中一张卡bs一致呢。如果是第二种情况,那么数据并行的总bs是比单卡大的,所以收敛会变慢,需要适当调整学习率
边栏推荐
- Golang logging library zerolog use record
- 深入浅出边缘云 | 3. 资源配置
- 云数据库和本地数据库有什么区别?
- crontab中写go run不执行的问题
- nlohmann json 使用指南【visual studio 2022】
- 【刷题篇】计算质数
- Basic use of scrapy
- 阿里云武林头条活动分享
- OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.解决方法
- Codeblocks + Widgets create window code analysis
猜你喜欢

6 yuan per catty, why do Japanese companies come to China to collect cigarette butts?

Scrapy框架介绍

VBA connects Access database and Excel

Mysql execution principle analysis

kotlin的by lazy

【剑指 Offer】剑指 Offer 22. 链表中倒数第k个节点

OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.解决方法

云数据库和本地数据库有什么区别?

谷歌AlphaFold近日宣称预测出地球上几乎所有蛋白质结构

nlohmann json 使用指南【visual studio 2022】
随机推荐
Object和Map的区别
解决终极bug,项目最终能顺利部署上线。
MindSpore: CV.Rescale(rescale,shift)中参数rescale和shift的含义?
Codeblocks + Widgets create window code analysis
Recommendation | People who are kind to you, don't repay them by inviting them to eat
Critical Reviews | 南农邹建文组综述全球农田土壤抗生素与耐药基因分布
6块钱1斤,日本公司为何来中国收烟头?
常见链表题及其 Go 实现
crontab中写go run不执行的问题
C# wpf 无边框窗口添加阴影效果
Basic use of scrapy
requet.getHeader(“token“) 为null
浅聊对比学习(Contrastive Learning)第一弹
golang日志库zerolog使用记录
运营 23 年,昔日“国内第一大电商网站”黄了...
The advanced version of the cattle brushing series (search for rotating sorted arrays, inversion of the specified range in the linked list)
【PHPWord】Quick Start of PHPWord in PHPOffice Suite
开心的聚餐
What is the value of biomedical papers? How to translate the papers into Chinese and English?
Go system collection