Rajut commited on
Commit
1bffeed
·
verified ·
1 Parent(s): 8caa9e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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': 6})
46
- model.train_model(adapted_data, num_train_epochs=6)
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