thibo73800 / capsnet-traffic-sign-classifier

A Tensorflow implementation of CapsNet(Capsules Net) apply on german traffic sign dataset
Apache License 2.0
177 stars 90 forks source link

How to draw the loss and accuracy diagrams with tensorboard? #8

Closed amirhfarzaneh closed 6 years ago

amirhfarzaneh commented 6 years ago

How do you use tensorboard to draw the loss and accuracy diagrams in the end?

Also how do you set an specific number of epochs for the algorithm to stop?

thibo73800 commented 6 years ago

Do you mean how to visualize the loss and accuracy?

To set a number of epochs, you need to edit the file train.py by replacing the "while True" line by something like this: "for e in range(epoch)"