fix(pt): typo in epoch training #14984

Job	Run time
Build source distribution	15s
Build wheels for cp311-macosx_arm64	3m 42s
Build wheels for cp310-manylinux_aarch64	4m 51s
Build wheels for cp311-win_amd64	5m 9s
Build wheels for cp311-manylinux_x86_64	6m 31s
Build wheels for cp311-macosx_x86_64	8m 57s
build_pypi_index	13s
build_docker (12)	6m 14s
upload_pypi	0s
Pass testing build wheels	3s
deploy_pypi_index	0s
	35m 55s

Provide feedback