Skip to content

Commit

Permalink
docs: train only for 8 epochs to speed up execution
Browse files Browse the repository at this point in the history
The result is not great with 32 epochs, either.
  • Loading branch information
lars-reimann committed Nov 26, 2024
1 parent fbf3973 commit 58b0067
Showing 1 changed file with 21 additions and 23 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,6 @@
},
{
"cell_type": "code",
"execution_count": null,
"id": "initial_id",
"metadata": {
"collapsed": true
Expand All @@ -57,7 +56,8 @@
"\n",
"images, filepaths = ImageList.from_files(\"data/shapes\", return_filenames=True)"
],
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -84,8 +84,8 @@
"collapsed": false
},
"id": "66dcf95a3fa51f23",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -108,8 +108,8 @@
"collapsed": false
},
"id": "32056ddf5396e070",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -149,8 +149,8 @@
"collapsed": false
},
"id": "806a8091249d533a",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -175,8 +175,8 @@
"collapsed": false
},
"id": "af68cc0d32655d32",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -198,15 +198,13 @@
},
{
"cell_type": "code",
"source": [
"cnn_fitted = cnn.fit(dataset, epoch_size=32, batch_size=16)"
],
"source": "cnn_fitted = cnn.fit(dataset, epoch_size=8, batch_size=16)",
"metadata": {
"collapsed": false
},
"id": "381627a94d500675",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -227,8 +225,8 @@
"collapsed": false
},
"id": "62f63dd68362c8b7",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -249,8 +247,8 @@
"collapsed": false
},
"id": "779277d73e30554d",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -271,8 +269,8 @@
"collapsed": false
},
"id": "a5ddbbfba41aa7f",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
},
{
"cell_type": "markdown",
Expand All @@ -293,8 +291,8 @@
"collapsed": false
},
"id": "7081595d7100fb42",
"execution_count": null,
"outputs": []
"outputs": [],
"execution_count": null
}
],
"metadata": {
Expand Down

0 comments on commit 58b0067

Please sign in to comment.