Spaces:
Build error
Build error
PeteBleackley
commited on
Commit
·
9946f4d
1
Parent(s):
7285c8a
Invoke class, not module
Browse files- scripts.py +5 -5
scripts.py
CHANGED
|
@@ -124,11 +124,11 @@ def train_models(path):
|
|
| 124 |
optimizer = keras.optimizers.Nadam(learning_rate=keras.optimizers.schedules.ExponentialDecay(1.0e-5, 100, 0.99))
|
| 125 |
trainer.compile(optimizer=optimizer,
|
| 126 |
loss=losses)
|
| 127 |
-
training_data = qarac.corpora.CombinedCorpus(tokenizer,
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
trainer.fit(training_data,
|
| 133 |
epochs=10,
|
| 134 |
workers=16,
|
|
|
|
| 124 |
optimizer = keras.optimizers.Nadam(learning_rate=keras.optimizers.schedules.ExponentialDecay(1.0e-5, 100, 0.99))
|
| 125 |
trainer.compile(optimizer=optimizer,
|
| 126 |
loss=losses)
|
| 127 |
+
training_data = qarac.corpora.CombinedCorpus.CombinedCorpus(tokenizer,
|
| 128 |
+
all_text='corpora/all_text.csv',
|
| 129 |
+
question_answering='corpora/question_answering.csv',
|
| 130 |
+
reasoning='corpora/reasoning_train.csv',
|
| 131 |
+
consistency='corpora/consistency.csv')
|
| 132 |
trainer.fit(training_data,
|
| 133 |
epochs=10,
|
| 134 |
workers=16,
|