按照官方的教程（https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.4/contrib/Matting/README_CN.md）进行训练，内存爆炸然后直接被kill了，请问怎么解决啊。谢谢

export CUDA_VISIBLE_DEVICES=0
python train.py \
--config configs/modnet/modnet_mobilenetv2.yml \
--do_eval \
--use_vdl \
--save_interval 5000 \
--num_workers 0 \ 
--save_dir output