Akegarasu / Akegarasu/lora-scripts

继续训练失败,KeyError: 'step'

オープン
#715 コメント 2 件 リアクション 0 件 担当者 0 名 GitHub で見る
主要言語
Python
スター
6.1k
フォーク
699
PR マージ指標
30日以内にマージされた PR はありません

説明

秋叶训练器训练lokr,设置为每步存一个档和一个模型,试过手动终止和强行终止,跑出的40多轮存档都报这个错,继续训练时也能读取 train_state.json 中的轮数和步数,训练参数未变动。
我不知道这个步骤读取的 'step' 起什么作用,总步数和当前步数CMD中都是正确显示的。

All model weights loaded successfully                                  checkpointing.py:214

2025-08-16 22:17:13 INFO     All optimizer states loaded successfully                               checkpointing.py:222

                    INFO     All scheduler states loaded successfully                               checkpointing.py:229

                    INFO     All dataloader sampler states loaded successfully                      checkpointing.py:241

                    INFO     GradScaler state loaded successfully                                   checkpointing.py:247

                    INFO     Could not load random states                                           checkpointing.py:264

Traceback (most recent call last):

  File "H:\M\machinelearning\sdtrainer\scripts\stable\sdxl_train_network.py", line 185, in

    trainer.train(args)

  File "H:\M\machinelearning\sdtrainer\scripts\stable\train_network.py", line 539, in train

    train_util.resume_from_local_or_hf_if_specified(accelerator, args)

  File "H:\M\machinelearning\sdtrainer\scripts\stable\library\train_util.py", line 4047, in resume_from_local_or_hf_if_specified

    accelerator.load_state(args.resume)

  File "H:\M\machinelearning\sdtrainer\python\Lib\site-packages\accelerate\accelerator.py", line 3156, in load_state

    self.step = override_attributes["step"]

                ~~~~~~~~~~~~~~~~~~~^^^^^^^^

KeyError: 'step'

22:17:15-724566 ERROR    Training failed / 训练失败

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。