Skip to content

Commit 96df168

Browse files
EddyLXJmeta-codesync[bot]
authored andcommitted
Fix changed name (#5138)
Summary: Pull Request resolved: #5138 X-link: meta-pytorch/torchrec#3549 X-link: https://github.com/facebookresearch/FBGEMM/pull/2140 is_st_publish is changed to load_ckpt_without_opt in diff D85830053 Reviewed By: emlin Differential Revision: D87135958 fbshipit-source-id: a693e766fe2c6cea58ad2ca1b4c6b80f54373564
1 parent 6f3d3b1 commit 96df168

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fbgemm_gpu/fbgemm_gpu/tbe/ssd/training.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def __init__(
249249
assert self.optimizer in [
250250
OptimType.EXACT_ROWWISE_ADAGRAD
251251
], f"only EXACT_ROWWISE_ADAGRAD supports embedding cache mode, but got {self.optimizer}"
252-
if self.is_st_publish:
252+
if self.load_ckpt_without_opt:
253253
if (
254254
# pyre-ignore [16]
255255
self.kv_zch_params.optimizer_type_for_st

0 commit comments

Comments
 (0)