Commit ff177bc
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,
),
Reviewed By: EddyLXJ
Differential Revision: D850171791 parent 8c83b41 commit ff177bc
File tree
1 file changed
+1
-0
lines changed- fbgemm_gpu/fbgemm_gpu
1 file changed
+1
-0
lines changedLines changed: 1 addition & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
241 | 241 | | |
242 | 242 | | |
243 | 243 | | |
| 244 | + | |
244 | 245 | | |
245 | 246 | | |
246 | 247 | | |
| |||
0 commit comments