Dusan Svilarkovic commited on
Commit
706cd84
1 Parent(s): 4da12b6
Files changed (1) hide show
  1. script.py +2 -1
script.py CHANGED
@@ -46,7 +46,8 @@ model_args.task = 'clickbait'
46
  model_args.device = device
47
  model_args.checkpoint = '../checkpoint-1464/'
48
 
49
- conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
 
50
  conditioning_model = conditioning_model.to(device)
51
  conditioning_model.eval()
52
 
 
46
  model_args.device = device
47
  model_args.checkpoint = '../checkpoint-1464/'
48
 
49
+ # conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
50
+ conditioning_model = Model(model_args, pad_id, vocab_size=None) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
51
  conditioning_model = conditioning_model.to(device)
52
  conditioning_model.eval()
53