removed printouts and fixed some code look
Merge branch 'New-gradient-batch' of https://github.com/johhor/id2210-vt14 into New-gradient-batch