Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Alescontrela committed Jun 16, 2018
1 parent 27f20a4 commit 5361b19
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,11 @@ Afterwards, you can train the network using the following command:

`python3 train_cnn.py '<file_name>.pkl'`

Replace <file_name> with whatever file name you would like. The terminal should display the following progress bar to indicate the training progress, as well as the cost for the current training batch.
Replace `<file_name>` with whatever file name you would like. The terminal should display the following progress bar to indicate the training progress, as well as the cost for the current training batch.

After the CNN has finished training, a .pkl file containing the network's parameters is saved to the directory where the script was run.

The network takes about 5 hours to train on my macbook pro. I included the trained params in the GitHub repo under the name params.pkl . To use them, replace <file_name> with params.pkl.
The network takes about 5 hours to train on my macbook pro. I included the trained params in the GitHub repo under the name `params.pkl` . To use the pretrained params when measuring the network's performance, replace `<file_name>` with params.pkl.

## Measuring Performance
To measure the network's accuracy, run the following command in the terminal:
Expand Down

0 comments on commit 5361b19

Please sign in to comment.