gaflow/MLproject at main · sungminwoo0612/gaflow · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
name: gaflow-yolo-experiments

conda_env: conda.yaml

entry_points:
  experiment_baseline:
    command: "python mlflow_experiments/exp_baseline.py"
    parameters:
      epochs:
        type: int
        default: 5
      batch_size:
        type: int
        default: 16
      model_size:
        type: str
        default: "yolov8n"

  experiment_hyperparams:
    command: "python mlflow_experiments/exp_hyperparams.py"
    parameters:
      epochs:
        type: int
        default: 10
      batch_size:
        type: int
        default: 32
      learning_rate:
        type: float
        default: 0.01
      model_size:
        type: str
        default: "yolov8s"

  experiment_model_size:
    command: "python mlflow_experiments/exp_model_size.py"
    parameters:
      epochs:
        type: int
        default: 5
      batch_size:
        type: int
        default: 16
      model_size:
        type: str
        default: "yolov8m"