Spaces:
Sleeping
Sleeping
run name as input
Browse files- user-friendly-metrics.py +5 -5
user-friendly-metrics.py
CHANGED
@@ -131,7 +131,7 @@ class UserFriendlyMetrics(evaluate.Metric):
|
|
131 |
formatted_datetime = current_datetime.strftime("%Y-%m-%d_%H-%M-%S")
|
132 |
wandb.login(key=os.getenv("WANDB_API_KEY"))
|
133 |
|
134 |
-
if
|
135 |
assert len(wandb_runs) == len(results), "runs and results must have the same length"
|
136 |
else:
|
137 |
wandb_runs = [f"{i}_{formatted_datetime}" for i in list(results.keys())]
|
@@ -145,10 +145,10 @@ class UserFriendlyMetrics(evaluate.Metric):
|
|
145 |
def wandb_run(self, result, wandb_run_name, wandb_project, debug):
|
146 |
|
147 |
run = wandb.init(
|
148 |
-
project=wandb_project,
|
149 |
-
name=
|
150 |
-
reinit=True,
|
151 |
-
settings=wandb.Settings(silent=not debug),
|
152 |
)
|
153 |
|
154 |
categories = {
|
|
|
131 |
formatted_datetime = current_datetime.strftime("%Y-%m-%d_%H-%M-%S")
|
132 |
wandb.login(key=os.getenv("WANDB_API_KEY"))
|
133 |
|
134 |
+
if wandb_runs is not None:
|
135 |
assert len(wandb_runs) == len(results), "runs and results must have the same length"
|
136 |
else:
|
137 |
wandb_runs = [f"{i}_{formatted_datetime}" for i in list(results.keys())]
|
|
|
145 |
def wandb_run(self, result, wandb_run_name, wandb_project, debug):
|
146 |
|
147 |
run = wandb.init(
|
148 |
+
project = wandb_project,
|
149 |
+
name = wandb_run_name,
|
150 |
+
reinit = True,
|
151 |
+
settings = wandb.Settings(silent=not debug),
|
152 |
)
|
153 |
|
154 |
categories = {
|