kangmingyu
Results
4
comments of
kangmingyu
我发现一个可能的解释了,那个FC层,AlexNet,VGG以及其他model,都使用了两层FC,每层4096个单元,占用了大量的内存。 but我感觉应该用不到这么多节点。
呃,不行,还是不够。。。默认的keras的数据类型是float32是吗? 我之前用tensorflow直接搭了个inception v4,299x299x3 size,也是float32,没有显存溢出问题。 这边代码应该是有点问题的。。
卧槽。。。。up主,你的VGG16模型那边,前两层卷积后,少了一次最大池化。。。。VGG19也是。
> adding encoding='utf-8' in open function when you open file NB!