Update train.py
Browse files
train.py
CHANGED
|
@@ -195,5 +195,5 @@ def train_model(config):
|
|
| 195 |
|
| 196 |
|
| 197 |
if __name__ == '__main__':
|
| 198 |
-
config = load_config(file_name='config.yaml')
|
| 199 |
train_model(config)
|
|
|
|
| 195 |
|
| 196 |
|
| 197 |
if __name__ == '__main__':
|
| 198 |
+
config = load_config(file_name='/config/config_final.yaml')
|
| 199 |
train_model(config)
|