Skip to content

Commit f8b9129

Browse files
emlinfacebook-github-bot
authored andcommitted
enable feature score auto collection in EBC (#5031)
Summary: X-link: meta-pytorch/torchrec#3475 X-link: facebookresearch/FBGEMM#2044 Enable feature score auto collection for EBC in the similar way of EC. The configuration has no difference in embedding table config: virtual_table_eviction_policy=FeatureScoreBasedEvictionPolicy( training_id_eviction_trigger_count=260_000_000, # 260M training_id_keep_count=160_000_000, # 160M enable_auto_feature_score_collection=True, feature_score_mapping={ "sparse_public_original_content_creator": 1.0, }, feature_score_default_value=0.5, ), Differential Revision: D85017179
1 parent b0dffd3 commit f8b9129

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,7 @@ class KVZCHParams(NamedTuple):
228228
backend_return_whole_row: bool = False
229229
eviction_policy: EvictionPolicy = EvictionPolicy()
230230
embedding_cache_mode: bool = False
231+
feature_score_collection_enabled: bool = False
231232

232233
def validate(self) -> None:
233234
assert len(self.bucket_offsets) == len(self.bucket_sizes), (

0 commit comments

Comments
 (0)