InternLM / InternLM/xtuner

一些关于步数统计的疑问

Open
#918 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
5.2k
Forks
448
Avg merge
3d 15h
Merged PRs (30d)
26

Description

在使用其它框架(Megatron-LM、DeepSpeed)训练的时候,一般统一把考虑梯度累积后的batch-size作为真正的单步batch-size,再根据这个batch-size推导训练步数,但是从xtuner训练时的步数显示来看,显然xtuner的步数逻辑不是这样的,这会导致一个问题,当我同时跑多个不同size模型的runs时,由于梯度累积值不同,所有的runs步数无法对齐,在wandb上看的体验很不好,请问有相关的设置可以保证我的步数为考虑梯度累积以后的步数吗?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.