Removed BaseNetwork making labels into kwargs