Spaces:
Runtime error
Runtime error
Commit
·
a36040c
1
Parent(s):
fd0734b
changed model name while logging to comet
Browse files- src/scripts/train.py +1 -1
src/scripts/train.py
CHANGED
@@ -57,7 +57,7 @@ def train(args):
|
|
57 |
# log model to comet
|
58 |
if "LOCAL_SYSTEM" not in os.environ:
|
59 |
if experiment:
|
60 |
-
experiment.log_model(f"
|
61 |
|
62 |
# evaluate model
|
63 |
print_title("\nEvaluating Model")
|
|
|
57 |
# log model to comet
|
58 |
if "LOCAL_SYSTEM" not in os.environ:
|
59 |
if experiment:
|
60 |
+
experiment.log_model(f"model",model_dir)
|
61 |
|
62 |
# evaluate model
|
63 |
print_title("\nEvaluating Model")
|