feat(tf2): support DPA4 training#5749

njzjz wants to merge 4 commits into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

njzjz:feat/dpa4-tf2-trainnjzjz/deepmd-kit:feat/dpa4-tf2-trainCopy head branch name to clipboard