chore: continued working on #36

logic/models/train/train.go

@@ -159,6 +159,7 @@ func trainDefinition(c *Context, model *BaseModel, definition_id string) (accura
 		"DataDir":   path.Join(getDir(), "savedData", model.Id, "data"),
 		"RunPath":   run_path,
 		"ColorMode": model.ImageMode,
+        "Model": model,
 	}); err != nil {
 		return
 	}
@@ -166,7 +167,7 @@ func trainDefinition(c *Context, model *BaseModel, definition_id string) (accura
 	// Run the command
 	out, err := exec.Command("bash", "-c", fmt.Sprintf("cd %s && python run.py", run_path)).Output()
 	if err != nil {
-		fmt.Println(string(out))
+        c.Logger.Debug(string(out))
 		return
 	}