我在我的笔记本上跑的时候，总是报内存溢出

我的命令是：python3 main.py --output_dir=model_output_0120 ==phase=train --training_set=.\datasets\tfrecords\VOC0712.tfrecords --training_steps=100000 --summary_steps=20 --checkpoint_steps=100 --save_steps=50
代码中的 batch_size 我改成 了 1，结果 c_step 跑到 2 的时候，每次都报 OOM……
不知道有什么办法能让 train 跑起来？