From 5361b19ef784088401c486a028c4f4f3bc0608c0 Mon Sep 17 00:00:00 2001 From: Alejandro Escontrela Date: Fri, 15 Jun 2018 21:20:39 -0400 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 8e7aa94..0e13fbd 100644 --- a/README.md +++ b/README.md @@ -17,11 +17,11 @@ Afterwards, you can train the network using the following command: `python3 train_cnn.py '.pkl'` -Replace 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 `` 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 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 `` with params.pkl. ## Measuring Performance To measure the network's accuracy, run the following command in the terminal: