Spaces:
Paused
Paused
Update train.py
Browse files
train.py
CHANGED
@@ -182,4 +182,4 @@ def train(
|
|
182 |
trainer.fit(model, datamodule=datamodule)
|
183 |
|
184 |
if __name__ == '__main__':
|
185 |
-
train('configs/44k/config.json', 'logs/
|
|
|
182 |
trainer.fit(model, datamodule=datamodule)
|
183 |
|
184 |
if __name__ == '__main__':
|
185 |
+
train('configs/44k/config.json', 'logs/44k')
|