如何在张量流中创建Resnet?

时间:2020-05-08 23:35:49

标签: python tensorflow neural-network conv-neural-network resnet

我已经使用tensorflow创建了一个convnet,但是最近了解了resnet,并且想知道如何将其添加到网络中。

当前这是我的网络,我想向其中添加8个resnet块:

def create_model(input_shape):
    input = keras.Input(i_shape)
    x = Conv2D(256, kernel_size=3, activation="relu")(input)
    x = MaxPooling2D((2, 2))(x)
    x = Dropout(0.2)(x)

    x = Flatten()(x)
    x = Dense(64, activation="relu")(x)

    output = Dense(action_space_size, activation="linear")(x)

    model = keras.Model(input, output)

    model.compile(optimizer=Adam(lr=0.001), loss="mse", metrics=["accuracy"])

0 个答案:

没有答案