arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-02-02 13:07:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Replace lr with learning_rate in Keras optimizers, fixes #456

Browse Source
This commit is contained in:
Aurélien Geron
2021-08-31 20:54:35 +12:00
parent 1568ac3b94
commit 108fe1fa53
10 changed files with 94 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
14_deep_computer_vision_with_cnns.ipynb
View File

@@ -1092,7 +1092,7 @@
"for layer in base_model.layers:\n",
" layer.trainable = False\n",
"\n",
"optimizer = keras.optimizers.SGD(lr=0.2, momentum=0.9, decay=0.01)\n",
"optimizer = keras.optimizers.SGD(learning_rate=0.2, momentum=0.9, decay=0.01)\n",
"model.compile(loss=\"sparse_categorical_crossentropy\", optimizer=optimizer,\n",
" metrics=[\"accuracy\"])\n",
"history = model.fit(train_set,\n",