Add possibility to configure node sizes separately for each layer