Results 4 issues of 何亮

这部分相关的设置和信息似乎没有,有什么好的预训练模型可以参考吗 谢谢

此处把多线程加速注释掉,似乎能解决问题

我的命令是: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 跑起来?