tiny icon indicating copy to clipboard operation
tiny copied to clipboard
verified publisher icon mlcommons

Confusing unused batch_size Variable in train_vww.py

Open colbybanbury opened this issue 2 years ago • 0 comments

The training script defines both BATCH_SIZE and batch_size but only uses BATCH_SIZE. I suggest we remove the unused batch_size to avoid confusion.

https://github.com/mlcommons/tiny/blob/3bca1a3adc7d9ec5186c0ff376715f2d3b19f382/benchmark/training/visual_wake_words/train_vww.py#L19

https://github.com/mlcommons/tiny/blob/3bca1a3adc7d9ec5186c0ff376715f2d3b19f382/benchmark/training/visual_wake_words/train_vww.py#L33

colbybanbury avatar Mar 08 '23 17:03 colbybanbury