improve perf train