[BUG] PT backend logger error: AttributeError: 'tuple' object has no attribute 'items'

[Yi-FanLi]

Bug summary

I am training a model with PyTorch backend. The setup is the same as what I used in the Tensorflow backend. However, it raises the AttributeError.

DeePMD-kit Version

3.0.0a0

Backend and its version

PyTorch v2.0.0.post200-gc263bd43e8e

How did you download the software?

docker

Input Files, Running Commands, Error Log, etc.

Error log:

[2024-05-11 04:58:48,328] DEEPMD INFO found 1 system(s):
[2024-05-11 04:58:48,328] DEEPMD INFO system natoms bch_sz n_bch prob pbc
[2024-05-11 04:58:48,328] DEEPMD INFO O64H128 192 1 100 1.000e+00 T
[2024-05-11 04:58:48,328] DEEPMD INFO --------------------------------------------------------------------------------------
[2024-05-11 04:58:48,331] DEEPMD INFO Start to train 400000 steps.
[2024-05-11 04:58:54,615] DEEPMD INFO batch 0: trn: rmse = 2.36e+01, rmse_e = 5.74e-01, rmse_f = 7.45e-01, lr = 1.00e-03
Traceback (most recent call last):
File "/opt/deepmd-kit/bin/dp", line 10, in
sys.exit(main())
^^^^^^
File "/opt/deepmd-kit/lib/python3.11/site-packages/deepmd/main.py", line 805, in main
deepmd_main(args)
File "/opt/deepmd-kit/lib/python3.11/site-packages/torch/distributed/elastic/multiprocessing/errors/init.py", line 346, in wrapper
return f(*args, **kwargs)
^^^^^^^^^^^^^^^^^^
File "/opt/deepmd-kit/lib/python3.11/site-packages/deepmd/pt/entrypoints/main.py", line 308, in main
train(FLAGS)
File "/opt/deepmd-kit/lib/python3.11/site-packages/deepmd/pt/entrypoints/main.py", line 280, in train
trainer.run()
File "/opt/deepmd-kit/lib/python3.11/site-packages/deepmd/pt/train/training.py", line 859, in run
step(step_id, model_key)
File "/opt/deepmd-kit/lib/python3.11/site-packages/deepmd/pt/train/training.py", line 757, in step
format_training_message_per_task(
File "/opt/deepmd-kit/lib/python3.11/site-packages/deepmd/loggers/training.py", line 29, in format_training_message_per_task
rmse = dict(sorted(rmse.items()))
^^^^^^^^^^
AttributeError: 'tuple' object has no attribute 'items'

Steps to Reproduce

See the tarball.

Further Information, Files, and Links

issue_logger.tar.gz

[njzjz]

I think this issue has been fixed. Could you try the latest commit?

[Yi-FanLi]

I confirm that the current devel branch does not have this issue.