manan commited on
Commit
06cefeb
1 Parent(s): 28b41a4

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -18,7 +18,7 @@ config = dict(
18
 
19
  # model info
20
  tokenizer_path = 'allenai/biomed_roberta_base', # 'roberta-base',
21
- model_checkpoint = '../input/biomed-roberta', # 'roberta-base',
22
  device = 'cuda' if torch.cuda.is_available() else 'cpu',
23
 
24
  # training paramters
 
18
 
19
  # model info
20
  tokenizer_path = 'allenai/biomed_roberta_base', # 'roberta-base',
21
+ model_checkpoint = 'allenai/biomed_roberta_base', # 'roberta-base',
22
  device = 'cuda' if torch.cuda.is_available() else 'cpu',
23
 
24
  # training paramters