train.py也报错

[LXX230]

哥，在s版本的时候，train.py在运行的时候总是在第4个epoch突然中断， tensorflow.python.framework.errors_impl.InvalidArgumentError: Incompatible shapes: [8400,5] vs. [0,5] [[{{node model_1/yolo_loss/while/body/_1/model_1/yolo_loss/while/logistic_loss_1/mul}}]] [Op:__inference_test_function_37457]

Function call stack: test_function

2021-11-18 20:07:59.290423: W tensorflow/core/kernels/data/generator_dataset_op.cc:107] Error occurred when finalizing GeneratorDataset iterator: Failed precondition: Python interpreter state is not initialized. The process may be terminated. [[{{node PyFunc}}]]

但是GPU还在被调用

[bubbliiiing]

我看看，好像还有个同学是这个问题，我看看咋回事

[bubbliiiing]

我复现不出来…………

[ZuyongWu]

我复现不出来…………

Incompatible shapes: [8400,5] vs. [0,5] ------ 我感觉主要是这个问题，改一下batch看看，我改了就没报错了

[LXX230]

我复现不出来…………

Incompatible shapes: [8400,5] vs. [0,5] ------ 我感觉主要是这个问题，改一下batch看看，我改了就没报错了

我3080只有10g显存，都把批次调到4一2了，调不了了，超显存了😩

[ZuyongWu]

我复现不出来…………

Incompatible shapes: [8400,5] vs. [0,5] ------ 我感觉主要是这个问题，改一下batch看看，我改了就没报错了

我3080只有10g显存，都把批次调到4一2了，调不了了，超显存了😩

调小一点。。。。。试试看

[bubbliiiing]

没道理啊，我是1660ti都不会炸显存，我用3060是8 4跑的，虽然是s

[ZuyongWu]

没道理啊，我是1660ti都不会炸显存，我用3060是8 4跑的，虽然是s

我用1660Ti跑s、l都没问题，3060跑s没问题，跑l报了这个错误，难以理解。。。。batch改到 4、2才正常

[LXX230]

没道理啊，我是1660ti都不会炸显存，我用3060是8 4跑的，虽然是s

我用1660Ti跑s、l都没问题，3060跑s没问题，跑l报了这个错误，难以理解。。。。batch改到 4、2才正常

我基本上，都是给设置成8一4，前50次基本上显存会达到9g左右了，后50次基本上就会超显存了，程序就停

[xddd315]

我也是这个问题，跑冻结的时候还撑得住，到了解冻环节第一个epoch 一半多就报这个错了。：）batch 已经调到2 2了。 3050Ti 4G显存，哭辽

[bubbliiiing]

说实话……我觉得一般是环境问题，环境问题会导致显存的错误

[xddd315]

说实话……我觉得一般是环境问题，环境问题会导致显存的错误 把图片大小调成（416.416）后，给跑通了。[doge]

[bubbliiiing]

0 0可以是可以，不过改小input_shape也会影响检测效果