@@ -272,7 +272,7 @@ LR climbs from warmup_initial_lr with even steps to initial lr. When warmup_init
 
 <dl class="py class">
 <dt class="sig sig-object py" id="super_gradients.training.utils.callbacks.YoloV5WarmupLRCallback">
-<em class="property"><span class="pre">class</span> </em><span class="sig-prename descclassname"><span class="pre">super_gradients.training.utils.callbacks.</span></span><span class="sig-name descname"><span class="pre">YoloV5WarmupLRCallback</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/sup
+<em class="property"><span class="pre">class</span> </em><span class="sig-prename descclassname"><span class="pre">super_gradients.training.utils.callbacks.</span></span><span class="sig-name descname"><span class="pre">YoloWarmupLRCallback</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/super
 <dd><p>Bases: <a class="reference internal" href="#super_gradients.training.utils.callbacks.LRCallbackBase" title="super_gradients.training.utils.callbacks.LRCallbackBase"><code class="xref py py-class docutils literal notranslate"><span class="pre">super_gradients.training.utils.callbacks.LRCallbackBase</span></code></a></p>
 </dd></dl>