Commit abc6d1f
enable feature score auto collection in EBC (pytorch#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: D850171791 parent 7f793b8 commit abc6d1f
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 | |
|---|---|---|---|
| |||
228 | 228 | | |
229 | 229 | | |
230 | 230 | | |
| 231 | + | |
231 | 232 | | |
232 | 233 | | |
233 | 234 | | |
| |||
0 commit comments