Установка скорости обучения в сверточной нейронной сети - PullRequest
1 голос
/ 18 октября 2019

Я использую Google Collab для изучения некоторых CNN.

Я использую model.compile (), чтобы установить функцию потерь и оптимизатора.

Где я могу изменить скорость обучения в следующих случаяхкод?

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

1 Ответ

2 голосов
/ 18 октября 2019

Вместо передачи строки вы можете передать оптимизатор для компиляции метода и установить скорость обучения для оптимизатора, как показано ниже:

from keras import optimizers
optm = optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=False)
model.compile(optimizer=optm,
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])
...