Not working for batch size=2 #10
Description
I am trying to experiment with batch size of 2 but seems like there is an index error in loss.py
{{ message }}
I am trying to experiment with batch size of 2 but seems like there is an index error in loss.py