fix

vms/ui/monitoring/tabs/__init__.py

@@ -3,7 +3,9 @@ Tab components for the "monitor" view
 """
 
 from .general_tab import GeneralTab
+from .gpu_tab import GPUTab
 
 __all__ = [
-    'GeneralTab'
+    'GeneralTab',
+    'GPUTab'
 ]

vms/ui/project/services/training.py

@@ -1150,7 +1150,7 @@ class TrainingService:
                 "learning_rate": params.get('learning_rate', DEFAULT_LEARNING_RATE),
                 "save_iterations": params.get('save_iterations', DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS),
                 "training_preset": params.get('preset_name', list(TRAINING_PRESETS.keys())[0]),
-                "auto_resume_checkbox": ui_state.get("auto_resume", DEFAULT_AUTO_RESUME)
+                "auto_resume_checkbox": params.get("auto_resume", DEFAULT_AUTO_RESUME)
             })
             
             # Check if we should auto-recover (immediate restart)