diff --git a/docs/transformers/compressive/experiment.html b/docs/transformers/compressive/experiment.html
index a57def7d..201f6a53 100644
--- a/docs/transformers/compressive/experiment.html
+++ b/docs/transformers/compressive/experiment.html
@@ -1242,7 +1242,7 @@ Memories that were compressed is needed for the reconstruction loss computation.
 <span class="lineno">335</span>                        <span class="s1">&#39;epochs&#39;</span><span class="p">:</span> <span class="mi">128</span><span class="p">,</span>
 <span class="lineno">336</span>                        <span class="s1">&#39;batch_size&#39;</span><span class="p">:</span> <span class="mi">32</span><span class="p">,</span>
 <span class="lineno">337</span>                        <span class="s1">&#39;inner_iterations&#39;</span><span class="p">:</span> <span class="mi">25</span><span class="p">,</span>
-<span class="lineno">338</span>                        <span class="s1">&#39;compression_ratio&#39;</span><span class="p">:</span> <span class="mi">2</span><span class="p">,</span>
+<span class="lineno">338</span>                        <span class="s1">&#39;compression_rate&#39;</span><span class="p">:</span> <span class="mi">2</span><span class="p">,</span>
 <span class="lineno">339</span>                        <span class="p">})</span></pre></div>
             </div>
         </div>
diff --git a/labml_nn/transformers/compressive/experiment.py b/labml_nn/transformers/compressive/experiment.py
index dd3af8e1..71d8ec64 100644
--- a/labml_nn/transformers/compressive/experiment.py
+++ b/labml_nn/transformers/compressive/experiment.py
@@ -335,7 +335,7 @@ def main():
                         'epochs': 128,
                         'batch_size': 32,
                         'inner_iterations': 25,
-                        'compression_ratio': 2,
+                        'compression_rate': 2,
                         })
 
     # Set models for saving and loading