Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
45e8538
1
Parent(s):
b13ce58
Update train_model.py
Browse files
src/models/train_model.py
CHANGED
@@ -30,6 +30,9 @@ def train_model():
|
|
30 |
|
31 |
model.save_model(model_dir=params['model_dir'])
|
32 |
|
|
|
|
|
|
|
33 |
|
34 |
if __name__ == '__main__':
|
35 |
train_model()
|
|
|
30 |
|
31 |
model.save_model(model_dir=params['model_dir'])
|
32 |
|
33 |
+
with open('wandb/latest-run/files/wandb-summary.json') as json_file:
|
34 |
+
data = json.load(json_file)
|
35 |
+
|
36 |
|
37 |
if __name__ == '__main__':
|
38 |
train_model()
|