修改部分代码，以便tim方便使用location的训练