s464980/IUM_s464980
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sacred fix 1
Some checks failed
s464980-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Sheaza 2024-06-09 18:05:59 +02:00
parent 7a3e24a150
commit 343ebbf307
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_sacred.py
View File

@ -54,7 +54,7 @@ class RegressionModel:
def save_model(self): def save_model(self):
self.model.save("model.keras") self.model.save("model.keras")
ex.add_artifact("model.keras") ex.add_artifact("model.keras", content_type="application/octet-stream")
@ex.main @ex.main