PeteBleackley commited on
Commit
9946f4d
·
1 Parent(s): 7285c8a

Invoke class, not module

Browse files
Files changed (1) hide show
  1. 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
- 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,
 
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,