首页 > 解决方案 > DL模型权重文件大的原因是什么?

问题描述

为什么以下简单(我猜)架构的模型权重为 950 MB,而 EfficientNetB5 架构的模型权重为 350MB。有什么问题?Conv2D 可能是 32?

model.add(Input(shape=(img_height, img_width, channels)))
                    model.add(Conv2D(32, (c2d, c2d), 
                                     padding="same", 
                                     use_bias = True, 
                                     activation="relu"))
                    model.add(MaxPooling2D(pool_size=(5, 5)))
                    model.add(Conv2D(16, (c2d, c2d), 
                                     padding="same", 
                                     use_bias = True, 
                                     activation="relu"))
                    model.add(MaxPooling2D(pool_size=(3, 3)))
                    model.add(Conv2D(8, (c2d, c2d), 
                                     padding="same", 
                                     use_bias = True, 
                                     activation="relu"))
                    model.add(MaxPooling2D(pool_size=(3, 3)))                    
                    model.add(Flatten())
                    model.add(Dense(ds, activation='relu'))
                    model.add(Dropout(do))
                    model.add(BatchNormalization())
                    model.add(Dense(2, activation='softmax'))

标签: kerasdeep-learning

解决方案


推荐阅读