Skip to content

Commit 8b4297f

Browse files
chore: Fix inaccurate documentation
1 parent 1be6ebb commit 8b4297f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ssd300_training.ipynb

+4-4
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,9 @@
163163
"\n",
164164
"model.load_weights(weights_path, by_name=True)\n",
165165
"\n",
166-
"# 3: Instantiate an Adam optimizer and the SSD loss function and compile the model.\n",
167-
"# If you want to follow the original Caffe implementation, use the SGD optimizer\n",
168-
"# that is commented out instead of the Adam optimizer.\n",
166+
"# 3: Instantiate an optimizer and the SSD loss function and compile the model.\n",
167+
"# If you want to follow the original Caffe implementation, use the preset SGD\n",
168+
"# optimizer, otherwise I'd recommend the commented-out Adam optimizer.\n",
169169
"\n",
170170
"#adam = Adam(lr=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-08, decay=0.0)\n",
171171
"sgd = SGD(lr=0.001, momentum=0.9, decay=0.0, nesterov=False)\n",
@@ -425,7 +425,7 @@
425425
"source": [
426426
"# Define model callbacks.\n",
427427
"\n",
428-
"# TODO: Set the filepath under which you want to save the weights.\n",
428+
"# TODO: Set the filepath under which you want to save the model.\n",
429429
"model_checkpoint = ModelCheckpoint(filepath='ssd300_pascal_07+12_epoch-{epoch:02d}_loss-{loss:.4f}_val_loss-{val_loss:.4f}.h5',\n",
430430
" monitor='val_loss',\n",
431431
" verbose=1,\n",

0 commit comments

Comments
 (0)