Skip to content

Commit 5f98e01

Browse files
EddyLXJmeta-codesync[bot]
authored andcommitted
Skip calculate feature score threshold for ttl in feature score eviction policy (#3552)
Summary: Pull Request resolved: #3552 AS title, no need to calculate feature score threshold for ttl in feature score eviction policy Reviewed By: steven1327 Differential Revision: D87278882 Privacy Context Container: L1158701 fbshipit-source-id: 4e1874b50de9c1f2bc28375d33a04770136e7cb4
1 parent 2bb54ba commit 5f98e01

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrec/modules/embedding_configs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ class FeatureScoreBasedEvictionPolicy(VirtualTableEvictionPolicy):
237237
0 # max number of inference ids per rank, default is training_id_keep_count
238238
)
239239
inference_eviction_feature_score_threshold: Optional[float] = (
240-
None # 0 means no eviction
240+
None # [DO NOT SET THIS] feature score threshold for inference eviction policy. This is used for inference only, and will be set during publish time
241241
)
242242
inference_eviction_ttl_mins: Optional[int] = None # 0 means no eviction
243243
feature_score_mapping: Optional[Dict[str, float]] = None # feature score mapping

0 commit comments

Comments
 (0)