From the code, it seems that the purpose of training is just to make the robot walk in a straight line. Shouldn't the result of training be speed tracking?

[yiyueming]

[linqi-ye]

It can be simply modified into speed tracking. Please refer to the KungfuSoccer example.