Update app.py
Browse files
app.py
CHANGED
@@ -42,8 +42,8 @@ model_args.n_best_size = 3
|
|
42 |
model_args.num_train_epochs = 5
|
43 |
|
44 |
# Model definition
|
45 |
-
model = QuestionAnsweringModel('bert', 'bert-base-uncased', use_cuda=False, args={'overwrite_output_dir': True, 'num_train_epochs':
|
46 |
-
model.train_model(adapted_data, num_train_epochs=
|
47 |
model.save_model(f"outputs/bert/final_model")
|
48 |
|
49 |
|
|
|
42 |
model_args.num_train_epochs = 5
|
43 |
|
44 |
# Model definition
|
45 |
+
model = QuestionAnsweringModel('bert', 'bert-base-uncased', use_cuda=False, args={'overwrite_output_dir': True, 'num_train_epochs': 4})
|
46 |
+
model.train_model(adapted_data, num_train_epochs=4)
|
47 |
model.save_model(f"outputs/bert/final_model")
|
48 |
|
49 |
|