启动模型训练

在容器中执行以下命令启动训练：

export ASCEND_RT_VISIBLE_DEVICES=0,1 
torchx run -s local_cwd dist.ddp -j 2 --script main.py