利用ray-train实现分布式训练