Skip to content

Commit

Permalink
Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
xiaohk committed Apr 30, 2020
1 parent ccb7071 commit 96dae87
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ Clone or download this repository:
```
git clone [email protected]:poloclub/cnn-explainer.git
# Or use degit, if you don't want to download commit histories
# Use degit if you don't want to download commit histories
degit poloclub/cnn-explainer
```

Expand All @@ -27,13 +27,13 @@ Install the dependencies:
npm install
```

Than run CNN Explainer:
Then run CNN Explainer:

```
npm run dev
```

Navigate to [localhost:5000](https://localhost:5000). You should CNN Explainer running in your broswer :)
Navigate to [localhost:5000](https://localhost:5000). You should see CNN Explainer running in your broswer :)

To see how we trained the CNN, visit the directory [`./tiny-vgg/`](tiny-vgg).

Expand Down
8 changes: 4 additions & 4 deletions tiny-vgg/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

This directory includes code and data to train a Tiny VGG model
(inspired by the demo CNN in [Stanford CS231n class](http:https://cs231n.stanford.edu))
on 10 every-day classes from the [Tiny ImageNet](https://tiny-imagenet.herokuapp.com).
on 10 everyday classes from the [Tiny ImageNet](https://tiny-imagenet.herokuapp.com).

## Installation

Expand Down Expand Up @@ -55,7 +55,7 @@ First, you want to unzip `data.zip`. The file structure would be something like:
To install all dependencies, run the following code

```
conda env create --file environment.yml
conda env create --file environment.yaml
```

## Training
Expand All @@ -66,14 +66,14 @@ To train Tiny VGG on these 10 classes, run the following code
python tiny-vgg.py
```

After training, you will get two trained models in Keras format `trained_tiny_vgg.h5`
After training, you will get two saved models in Keras format: `trained_tiny_vgg.h5`
and `trained_vgg_best.h5`. The first file is the final model after training, and
`trained_vgg_best.h5` is the model having the best validation performance.
You can use either one for CNN Explainer.

## Convert Model Format

Before loading the model using *tensorflow.js*, you want to convert the model format
Before loading the model using *tensorflow.js*, you want to convert the model file
from Keras `h5` format to [tensorflow.js format](https://www.tensorflow.org/js/tutorials/conversion/import_keras).

```
Expand Down

0 comments on commit 96dae87

Please sign in to comment.