add support for node weighted loss augmention