a b/consts.py 

  1
  
import torch





  
  

  2
  





  
  

  3
  
MAX_SENTENCE_LENGTh = 70





  
  

  4
  
device = "cuda" if torch.cuda.is_available() else "cpu"





  
  

  5
  
model_name = 'microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext' #('bert-base-uncased')   'microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext'