diff --git a/scar/test/test_scar.py b/scar/test/test_scar.py index 502d62d..3080d05 100755 --- a/scar/test/test_scar.py +++ b/scar/test/test_scar.py @@ -22,7 +22,7 @@ def test_scar(self): feature_type="sgRNAs", ) - scarObj.train(epochs=40, batch_size=64) + scarObj.train(epochs=40, batch_size=32) scarObj.inference() @@ -58,7 +58,7 @@ def test_scar_citeseq(self): feature_type="ADTs", ) - citeseq_scar.train(epochs=200, batch_size=64, verbose=False) + citeseq_scar.train(epochs=200, batch_size=32, verbose=False) citeseq_scar.inference() dist = euclidean_distances(