debug: add print for neurons per rank