Updated stylyze plan to connect weights to loss layers.