yolov4-tf2 icon indicating copy to clipboard operation
yolov4-tf2 copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

Resource exhausted

Open 6Lackiu opened this issue 3 years ago • 3 comments

我训练到大概第30个epoch时出现了这个问题，请问是爆显存了吗？ 我是在colab上进行训练的，分配的GPU是tesla T4，我设置的batch size是16. error1 error

6Lackiu avatar Apr 08 '21 04:04 6Lackiu

另外博主我想问一下，目前我的logs文件夹下有这些权重文件，如果我想断点继续训练的话，是将model_path指向哪个权值文件的路径呢？是ep032还是trained_weights_stage_1呢？并且这个trained_weights_stage_1.h5是什么权重文件呢？期待您的回答 谢谢！ weights

6Lackiu avatar Apr 08 '21 04:04 6Lackiu

1、是的 2、两个是同一个权值应该，trained_weights_stage_1.h5是冻结训练完保存的文件，你应该是解冻训练爆显存了，所以两个一样

bubbliiiing avatar Apr 15 '21 10:04 bubbliiiing

1、是的 2、两个是同一个权值应该，trained_weights_stage_1.h5是冻结训练完保存的文件，你应该是解冻训练爆显存了，所以两个一样

好的谢谢

6Lackiu avatar Apr 16 '21 14:04 6Lackiu