diff --git a/website/assets/img/docs/training-loop.svg b/website/assets/img/docs/training-loop.svg
index c0acd10cf..e670f816a 100644
--- a/website/assets/img/docs/training-loop.svg
+++ b/website/assets/img/docs/training-loop.svg
@@ -1,4 +1,4 @@
-<svg xmlns="http://www.w3.org/2000/svg" width="612" height="330" viewBox="-10 -10 622 360">
+<svg xmlns="http://www.w3.org/2000/svg" width="612" height="280" viewBox="-10 -10 622 360">
     <style>
         .svg__trainloop__text { fill: #1a1e23; font: 18px "Source Sans Pro", Tahoma, Helvetica, Arial, sans-serif }
         .svg__trainloop__text-small { fill: #1a1e23; font: 16px "Source Sans Pro", Tahoma, Helvetica, Arial, sans-serif }