Skip to content

Commit

Permalink
Docs: Use ImageNet dataset version 5.0.0 (#53)
Browse files Browse the repository at this point in the history
The default version of ImageNet includes `.png` images that are not correctly converted to `.jpeg` images. This will fail with the new preprocessing introduced in #52

Fixes larq/larq#174 (comment)
  • Loading branch information
lgeiger authored and jamescook106 committed Aug 8, 2019
1 parent cbe657d commit 61bb10e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/index.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,13 +24,13 @@ Larq Zoo ships with a command line interface powered by [`zookeeper`](https://gi
E.g. to reproduce the training of [Binary AlexNet](/models/api/#binaryalexnet) run:

```shell
lqz train binary_alexnet --dataset imagenet2012
lqz train binary_alexnet --dataset imagenet2012 --dataset-version 5.0.0
```

To experiment with different hyperparameters you can either edit the [`HParams` for this model](https://github.com/larq/zoo/blob/master/larq_zoo/binarynet.py#L72-L85) or overwrite them from the command line, e.g.:

```shell
lqz train binary_alexnet --dataset imagenet2012 --hparams epochs=150,batch_size=256
lqz train binary_alexnet --dataset imagenet2012 --dataset-version 5.0.0 --hparams epochs=150,batch_size=256
```

To view a TensorBoard for the current training, replace the `lqz train` command with `lqz tensorboard`.
Expand Down

0 comments on commit 61bb10e

Please sign in to comment.