Update README.md

ndrplz · Oct 15, 2017 · 0c994bf · 0c994bf
1 parent 1c43700
commit 0c994bf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/capstone_traffic_light_classifier/README.md b/capstone_traffic_light_classifier/README.md
@@ -10,4 +10,6 @@ The traffic light dataset is modeled in [`traffic_light_dataset.py`](https://git
 
 The deep network architecture is in [`traffic_light_classifier.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/traffic_light_classifier.py)
 
-Training can be started from [`main.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/main.py)
+Training the model: [`train.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/train.py)
+
+Testing the model from pretrained weights: [`test.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/test.py)