Skip to content

Commit

Permalink
Small update to the notebook.
Browse files Browse the repository at this point in the history
  • Loading branch information
drewoldag committed Oct 8, 2024
1 parent 665f04e commit b98d8b8
Showing 1 changed file with 5 additions and 9 deletions.
14 changes: 5 additions & 9 deletions docs/notebooks/TrainingAModel.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,8 @@
"metadata": {},
"outputs": [],
"source": [
"fibad_instance = fibad.Fibad(config_file=\"../../src/fibad/fibad_default_config.toml\")"
"# Create an instance of a fibad object, instantiated (implicitly) with the default configuration file\n",
"fibad_instance = fibad.Fibad()"
]
},
{
Expand All @@ -24,9 +25,10 @@
"metadata": {},
"outputs": [],
"source": [
"# Update a few of the configuration parameters\n",
"fibad_instance.config[\"model\"][\"name\"] = \"ExampleCNN\"\n",
"fibad_instance.config[\"data_set\"][\"name\"] = \"CifarDataSet\"\n",
"fibad_instance.config[\"data_loader\"][\"num_workers\"] = 1"
"fibad_instance.config[\"data_loader\"][\"batch_size\"] = 64"
]
},
{
Expand All @@ -35,15 +37,9 @@
"metadata": {},
"outputs": [],
"source": [
"# Begin training the Example CNN model using the CIFAR-10 dataset\n",
"fibad_instance.train()"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
Expand Down

0 comments on commit b98d8b8

Please sign in to comment.