efficientdet-keras icon indicating copy to clipboard operation
efficientdet-keras copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

freeze_layers设置的问题

Open OkuChou opened this issue 5 years ago • 1 comments

你好。 你设置的freeze_layers是[226,328,328,373,463,565,655,802] 我想问一下这样设置的理由是什么？ 而且感觉这样设置好像很吃显存，我显卡12G的，用b3进行训练，batch分别是4和2。50epoch前训练没问题，过了50ephoch后就爆掉了。请问是和这个设置freeze_layers有关系吗？是不是该设小一点呢？

OkuChou avatar Sep 04 '20 10:09 OkuChou

1、你可以数数每个网络的层数，冻结到主干特征提取网络 2、efficientdet本就很吃显存。12G也吃不消

bubbliiiing avatar Sep 05 '20 14:09 bubbliiiing