Commit Graph

21 Commits

Author SHA1 Message Date
VictorSanh
a1af5247e1 Add seed in initialization 2018-11-03 14:00:36 -04:00
thomwolf
25f73add07 update optimizer run_squad 2018-11-03 17:56:34 +01:00
thomwolf
f514cbbf30 update run_squad with tqdm 2018-11-03 17:52:44 +01:00
thomwolf
cb76c1ddd3 add model.zero_grad() 2018-11-03 17:40:12 +01:00
VictorSanh
8bd6b235b7 typo on tokenization 2018-11-03 10:27:59 -04:00
VictorSanh
5f432480c0 Create DataParallel model if several GPUs 2018-11-03 10:10:01 -04:00
VictorSanh
38f740a1d5 Fix bug writing predictions in run_squad_pytorch 2018-11-02 11:29:45 -04:00
VictorSanh
ee29871f8d Debug run_squad_pytorch 2018-11-02 11:07:32 -04:00
VictorSanh
101eabff90 Debug run_squad_pytorch 2018-11-02 10:44:08 -04:00
VictorSanh
bb0a510330 Print for debug run_squad 2018-11-02 10:16:07 -04:00
Tim Rault
3ff2ec5eb3 Move command-line argparse arguments into main() function 2018-11-02 14:42:05 +01:00
Tim Rault
49006d0662 Bug fix type=bool -> action='store_true' in argparse 2018-11-02 10:04:41 +01:00
VictorSanh
beb59080b3 Fix size compatibility for model.forward
Error was coming from "modeling_pytorch.py", line 484, in forward: start_loss = loss_fct(start_logits, start_positions) --> ValueError: Expected target size (12, 1), got torch.Size([12])
2018-11-02 04:36:42 -04:00
VictorSanh
8cbe7d6af8 FIX errors in loading eval Dataset in run_squad_pytorch 2018-11-02 04:09:53 -04:00
VictorSanh
833c3a7a25 FIX errors in loading Dataset in run_squad_pytorch 2018-11-02 04:00:00 -04:00
VictorSanh
62ac7e9a60 Fix small bug in run_squad_pytorch.py 2018-11-02 03:32:35 -04:00
thomwolf
ebfffa0ab2 updated extract_features 2018-11-02 04:32:18 +01:00
thomwolf
9af479b3b9 conversion run_squad ok 2018-11-02 04:12:20 +01:00
thomwolf
8e81e5e6ff working on squad 2018-11-02 04:07:52 +01:00
thomwolf
e61db0d1c0 run_squad WIP 2018-11-02 03:56:14 +01:00
Tim Rault
371945cf07 Convert flags to argparse in 'run_squad_pytorch.py' 2018-11-01 11:17:49 +01:00