reduce CI memory requirements

graphcore-research · Oct 24, 2023 · fbe91e9 · fbe91e9
1 parent da0ca26
commit fbe91e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
@@ -14,10 +14,10 @@
 from besskge.sharding import PartitionedTripleSet, Sharding
 
 seed = 1234
-n_entity = 20000
+n_entity = 5000
 n_relation_type = 50
 n_shard = 4
-n_test_triple = 5000
+n_test_triple = 1000
 batches_per_step = 2
 shard_bs = 400
 embedding_size = 128