for batch size=1024, loss stuck at 7
for batch size=1024, loss stuck at 7