Compare commits
No commits in common. "fb944fa73eca47872912fabf3d343b720e167a78" and "b679e7d9c66d6214cdbcdaa411812aeba7d6fd76" have entirely different histories.
fb944fa73e
...
b679e7d9c6
13
MLproject
13
MLproject
@ -1,13 +0,0 @@
|
||||
name: s424714
|
||||
|
||||
conda_env: environment.yml #ścieżka do pliku conda.yaml z definicją środowiska
|
||||
|
||||
entry_points:
|
||||
main:
|
||||
parameters:
|
||||
epochs: {type: int, default: 3}
|
||||
lr: {type: float, default: 0.000001}
|
||||
command: "python src/main.py --train --lr={lr} -e={epochs}"
|
||||
# command: "python src/train.py --train --lr={lr} -e={epochs}"
|
||||
validate:
|
||||
command: "python src/main.py --train"
|
@ -28,7 +28,7 @@ parser.add_argument("--num_epochs", "--epochs", "-e", type=int, default=3)
|
||||
|
||||
# mlflow stuff
|
||||
mlflow.set_tracking_uri("http://localhost:5000")
|
||||
# mlflow.set_experiment("s424714")
|
||||
mlflow.set_experiment("s424714")
|
||||
|
||||
|
||||
def main():
|
||||
|
Loading…
Reference in New Issue
Block a user