Как создать сетку в тензорном потоке? - PullRequest
Новая
       108

Как создать сетку в тензорном потоке?

2 голосов
/ 09 мая 2020

Я создал con vnet, используя тензорный поток, но недавно узнал о реснетах и ​​задавался вопросом, как я могу добавить его в свою сеть.

Сейчас это моя сеть, и я хотел бы добавить в нее 8 блоков re snet: 

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"])
...