Skip to content

Commit 25907c2

Browse files
authored
Fix README for fine-tuning and tutorial typo (#1092)
* Fix README for finetuning * Fix tutorial notebook
1 parent be5340b commit 25907c2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

efficientnetv2/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ You can directly use this code to build a model like this:
8282

8383
mode = tf.keras.models.Sequential([
8484
tf.keras.layers.InputLayer(input_shape=[224, 224, 3]),
85-
effnetv2_model.get_model('efficientnetv2-b0', include_top=False, pretrained=True),
85+
effnetv2_model.get_model('efficientnetv2-b0', include_top=False),
8686
tf.keras.layers.Dropout(rate=0.2),
8787
tf.keras.layers.Dense(4, activation='softmax'),
8888
])

efficientnetv2/tutorial.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@
181181
"pred = tf.keras.layers.Softmax()(logits)\n",
182182
"idx = tf.argsort(logits[0])[::-1][:5].numpy()\n",
183183
"import ast\n",
184-
"classes = ast.literal_eval(open(labels_file, \"r\").read())\n",
184+
"classes = ast.literal_eval(open(labels_map, \"r\").read())\n",
185185
"for i, id in enumerate(idx):\n",
186186
" print(f'top {i+1} ({pred[0][id]*100:.1f}%): {classes[id]} ')\n",
187187
"from IPython import display\n",

0 commit comments

Comments
 (0)