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
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.
tiny/benchmark/training/visual_wake_words/train_vww.py
Line 19 in 3bca1a3
tiny/benchmark/training/visual_wake_words/train_vww.py
Line 33 in 3bca1a3