cjlin1 / simpleNN

BSD 3-Clause "New" or "Revised" License
48 stars 16 forks source link

Use Tensorflow to calculate mean loss instead #4

Closed Sinacam closed 4 years ago

Sinacam commented 4 years ago

Previously calculates mean loss by dividing total loss by batch size manually, now uses tf.reduce_mean. This allows batch size to be aribtrarily varied. This doesn't affect current behavior.