Soumic commited on
Commit
3b659a2
·
1 Parent(s): 19c8c1e

:hammer: Reduce seq_len = 1000, batch_size = 16

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -489,7 +489,7 @@ def start_bert(classifier_model, criterion, m_optimizer=torch.optim.Adam, WINDOW
489
  if __name__ == '__main__':
490
  login_inside_huggingface_virtualmachine()
491
 
492
- WINDOW = 4000
493
  some_model = DnaBert6MQTLClassifier(seq_len=WINDOW, model_repository_name="dnabert-6-mqtl-classifier")
494
  criterion = ReshapedBCEWithLogitsLoss()
495
 
@@ -499,6 +499,6 @@ if __name__ == '__main__':
499
  WINDOW=WINDOW,
500
  is_debug=False,
501
  max_epochs=20,
502
- batch_size=32
503
  )
504
  pass
 
489
  if __name__ == '__main__':
490
  login_inside_huggingface_virtualmachine()
491
 
492
+ WINDOW = 1000
493
  some_model = DnaBert6MQTLClassifier(seq_len=WINDOW, model_repository_name="dnabert-6-mqtl-classifier")
494
  criterion = ReshapedBCEWithLogitsLoss()
495
 
 
499
  WINDOW=WINDOW,
500
  is_debug=False,
501
  max_epochs=20,
502
+ batch_size=16
503
  )
504
  pass