Fix "Start Training" button label (#116)

Fixes the "Start Button" in `widget_optimization` to show the correct label when loading the app window (introduced after a recent fix to the iterations counter in trainer). Signed-off-by: operel <[email protected]>
NVIDIAGameWorks · Feb 6, 2023 · 30abf8f · 30abf8f
1 parent d264bb9
commit 30abf8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wisp/renderer/gui/imgui/widget_optimization.py b/wisp/renderer/gui/imgui/widget_optimization.py
@@ -43,7 +43,7 @@ def paint(self, state: WispState, *args, **kwargs):
                 state.optimization.running = False
                 state.renderer.background_tasks_paused = True
         else:
-            if curr_epoch == 1 and curr_iteration == 1:
+            if curr_epoch == 1 and curr_iteration == 0:
                 if imgui.button("Start Training", width=button_width):
                     state.optimization.running = True
                     state.renderer.background_tasks_paused = False