Rajut commited on
Commit
b5ada77
·
verified ·
1 Parent(s): 6ffe34a

Update app.py

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