fixing batch size, updating logging