:hammer: Save progress

README.md

@@ -2,7 +2,7 @@
 title: Mqtl Classification Using Dnabert 6
 emoji: 👁
 colorFrom: blue
-colorTo: white
+colorTo: purple
 sdk: docker
 pinned: false
 license: creativeml-openrail-m

app.py

@@ -401,7 +401,7 @@ class DnaBert6MQTLClassifier(nn.Module):
 
 
 def start_bert(classifier_model, criterion, m_optimizer=torch.optim.Adam, WINDOW=200,
-          is_binned=True, is_debug=False, max_epochs=10, batch_size=8):
+               is_binned=True, is_debug=False, max_epochs=10, batch_size=8):
   file_suffix = ""
   if is_binned:
     file_suffix = "_binned"
@@ -488,9 +488,16 @@ def start_bert(classifier_model, criterion, m_optimizer=torch.optim.Adam, WINDOW
 if __name__ == '__main__':
   login_inside_huggingface_virtualmachine()
 
-  WINDOW = 200
+  WINDOW = 4000
   some_model = DnaBert6MQTLClassifier(seq_len=WINDOW, model_repository_name="dnabert-6-mqtl-classifier")
   criterion = ReshapedBCEWithLogitsLoss()
 
-  start_bert(classifier_model=some_model, criterion=criterion, WINDOW=WINDOW, is_debug=True, max_epochs=2)
+  start_bert(
+    classifier_model=some_model,
+    criterion=criterion,
+    WINDOW=WINDOW,
+    is_debug=False,
+    max_epochs=20,
+    batch_size=32
+  )
   pass