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 |
|