Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
9a78cd2
1
Parent(s):
cf780e0
Update train_model.py
Browse files
src/models/train_model.py
CHANGED
@@ -25,7 +25,6 @@ def train_model():
|
|
25 |
model_type=params["model_type"], model_name=params["model_name"]
|
26 |
)
|
27 |
|
28 |
-
num_workers=int(params['num_workers']))
|
29 |
|
30 |
model.save_model(model_dir=params['model_dir'])
|
31 |
|
|
|
25 |
model_type=params["model_type"], model_name=params["model_name"]
|
26 |
)
|
27 |
|
|
|
28 |
|
29 |
model.save_model(model_dir=params['model_dir'])
|
30 |
|