nickil commited on
Commit
f58f7f4
1 Parent(s): 245d478

update model ckpt

Browse files
weakly_supervised_parser/utils/populate_chart.py CHANGED
@@ -28,7 +28,7 @@ ptb_most_common_first_token = "the"
28
 
29
  from pytorch_lightning import Trainer
30
 
31
- trainer = Trainer(accelerator="auto", enable_progress_bar=False, gpus=-1)
32
 
33
 
34
  class PopulateCKYChart:
 
28
 
29
  from pytorch_lightning import Trainer
30
 
31
+ trainer = Trainer(accelerator="auto", enable_progress_bar=False)
32
 
33
 
34
  class PopulateCKYChart: