Skip to content

Commit 0b2e410

Browse files
committed
Adjusting commented lines
1 parent bd8d4a4 commit 0b2e410

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

dlrm_s_pytorch.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1572,11 +1572,10 @@ def run():
15721572

15731573
# # print("res: ", S)
15741574

1575-
# # print("j, train: BCE, shifted_BCE ", j, L, L_shifted)
1575+
# # print("j, train: BCE ", j, L)
15761576

15771577
# mbs = T.shape[0] # = args.mini_batch_size except maybe for last
15781578
# A = np.sum((np.round(S, 0) == T).astype(np.uint8))
1579-
# A_shifted = np.sum((np.round(S_shifted, 0) == T).astype(np.uint8))
15801579

15811580
with record_function("DLRM backward"):
15821581
# scaled error gradient propagation

0 commit comments

Comments
 (0)