Spaces:
Runtime error
Runtime error
Soumic
commited on
Commit
·
3b659a2
1
Parent(s):
19c8c1e
:hammer: Reduce seq_len = 1000, batch_size = 16
Browse files
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 =
|
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=
|
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
|