refactor batch_size->batch_len