fix stylegan torgb bias

Author: vpj

docs/gan/stylegan/index.html

@@ -889,7 +889,7 @@ <h3>To RGB</h3>
                 <p>Bias</p>
             </div>
             <div class='code'>
-                <div class="highlight"><pre><span class="lineno">402</span>        <span class="bp">self</span><span class="o">.</span><span class="n">bias</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Parameter</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">zeros</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span></pre></div>
+                <div class="highlight"><pre><span class="lineno">402</span>        <span class="bp">self</span><span class="o">.</span><span class="n">bias</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Parameter</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">zeros</span><span class="p">(</span><span class="mi">3</span><span class="p">))</span></pre></div>
             </div>
         </div>
     <div class='section' id='section-59'>

labml_nn/gan/stylegan/__init__.py

@@ -399,7 +399,7 @@ def __init__(self, d_latent: int, features: int):
         # Weight modulated convolution layer without demodulation
         self.conv = Conv2dWeightModulate(features, 3, kernel_size=1, demodulate=False)
         # Bias
-        self.bias = nn.Parameter(torch.zeros(1))
+        self.bias = nn.Parameter(torch.zeros(3))
         # Activation function
         self.activation = nn.LeakyReLU(0.2, True)