added analytics and logs

api/src/yolo/config.py

@@ -35,6 +35,10 @@ class Config:
     model_name: str = "face_detection_yolo"
     save: bool = True
 
+    # Run log (src/common/run_log.py): one folder per training run
+    runs_dir: str = "runs"            # empty = no run log
+    run_name: str = ""                # added to the run folder name, e.g. "bn-lr3e-4"
+
     # sample / webcam tasks
     checkpoint: str = ""              # empty = <save_dir>/<model_name>/<model_name>
     threshold: float = 0.95           # confidence needed to draw a box

api/src/yolo/tasks.py

@@ -62,7 +62,7 @@ def train(cfg: Config):
 
     fit(model, YoloLoss(), optimizer, train_loader, val_loader, device,
         epochs=cfg.epochs, save_dir=cfg.save_dir, model_name=cfg.model_name, save=cfg.save,
-        monitor="loss")
+        monitor="loss", config=cfg)
 
 
 def view_data(cfg: Config):