undertherain / benchmarker

modular framework for [not only] deep learning performance benchmarking
http://blackbird.pw/performance
Mozilla Public License 2.0
9 stars 5 forks source link

Fix (mini fix) pytorch output #67

Closed vatai closed 4 years ago

vatai commented 4 years ago

batch_idx is multiplied by len(data) (which results in the number of samples processed). This doesn't match the next value after '/', is len(self.x_train) which is the total number of batches (and not total number of samples).

This closes #66