rec_pangu.benchmark_trainer