decrease batch size to 10