partitioning
Browse files
finetune_categorisation_xxl.gin
CHANGED
@@ -36,5 +36,5 @@ utils.SaveCheckpointConfig:
|
|
36 |
# LOSS_NORMALIZING_FACTOR = 234496
|
37 |
|
38 |
# Might have to ba changed based on architecture
|
39 |
-
|
40 |
|
|
|
36 |
# LOSS_NORMALIZING_FACTOR = 234496
|
37 |
|
38 |
# Might have to ba changed based on architecture
|
39 |
+
partitioning.PjitPartitioner.num_partitions = 4
|
40 |
|