Skip to content

Commit

Permalink
Add comment about dataset structure
Browse files Browse the repository at this point in the history
  • Loading branch information
timesler committed Sep 10, 2019
1 parent d0f6fba commit e644fcc
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion examples/finetune.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,9 @@
"metadata": {},
"outputs": [],
"source": [
"data_dir = '/mnt/windows/Users/times/Data/vggface2/test'\n",
"data_dir = '/path/to/dataset'\n",
"# Dataset should follow the VGGFace2/ImageNet-style directory layout\n",
"\n",
"batch_size = 32\n",
"epochs = 15\n",
"workers = 4"
Expand Down

0 comments on commit e644fcc

Please sign in to comment.