Skip to content

Commit b85fb70

Browse files
authored
Update __init__.py
1 parent 2ebd980 commit b85fb70

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

api/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -166,9 +166,9 @@ def train(**args):
166166
settings.update(
167167
{
168168
"mlflow": False,
169-
"datasets_dir": config.DATA_PATH,
170-
"model_dir": config.MODELS_PATH,
171-
"wandb": str(args["disable_wandb"]),
169+
#"datasets_dir": config.DATA_PATH,
170+
# "model_dir": config.MODELS_PATH,
171+
"wandb": args["disable_wandb"],
172172
}
173173
)
174174
# Modify the model name based on task type

0 commit comments

Comments
 (0)