bugfix
Browse files- train.ipynb +2 -0
train.ipynb
CHANGED
|
@@ -418,6 +418,8 @@
|
|
| 418 |
"\n",
|
| 419 |
" return history_fine\n",
|
| 420 |
"\n",
|
|
|
|
|
|
|
| 421 |
"model = create_efficientnet_model((64, 64, 3), len(class_names))\n",
|
| 422 |
"\n",
|
| 423 |
"model.compile(optimizer='adam',\n",
|
|
|
|
| 418 |
"\n",
|
| 419 |
" return history_fine\n",
|
| 420 |
"\n",
|
| 421 |
+
"train_ds = augment_data(train_ds)\n",
|
| 422 |
+
"\n",
|
| 423 |
"model = create_efficientnet_model((64, 64, 3), len(class_names))\n",
|
| 424 |
"\n",
|
| 425 |
"model.compile(optimizer='adam',\n",
|