Fix visualization

rohithkrn · Aug 6, 2019 · 785ac51 · 785ac51
1 parent e8c77b9
commit 785ac51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dlrm_s_pytorch.py b/dlrm_s_pytorch.py
@@ -887,8 +887,8 @@ def loss_fn_wrap(Z, T, use_gpu, device):
             + " visualization. Then, uncomment its import above as well as"
             + " three lines below and run the code again."
         )
-        # V = Z.mean() if args.inference_only else L
-        # make_dot(V, params=dict(dlrm.named_parameters()))
+        # V = Z.mean() if args.inference_only else E
+        # dot = make_dot(V, params=dict(dlrm.named_parameters()))
         # dot.render('dlrm_s_pytorch_graph') # write .pdf file
 
     # test prints