linting

autonlab · Dec 28, 2020 · 9ea6c2c · 9ea6c2c
1 parent 569e79b
commit 9ea6c2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dsm/dsm_torch.py b/dsm/dsm_torch.py
@@ -473,7 +473,9 @@ def __init__(self, inputdim, k, typ='ConvNet',
         nn.Linear(hidden, k, bias=True)
         ) for r in range(self.risks)})
 
-    self.embedding = create_conv_representation(inputdim=inputdim, hidden=hidden, typ='ConvNet')
+    self.embedding = create_conv_representation(inputdim=inputdim, 
+                                                hidden=hidden, 
+                                                typ='ConvNet')
 
   def forward(self, x, risk='1'):
     """The forward function that is called when data is passed through DSM.

diff --git a/dsm/utilities.py b/dsm/utilities.py
@@ -165,7 +165,7 @@ def train_dsm(model,
                                  _reshape_tensor_with_nans(eb),
                                  elbo=elbo,
                                  risk=str(r+1))
-      print ("Train Loss:", float(loss), "batch num:", j, "/", nbatches, "n_iter:", i, "/", n_iter)
+      #print ("Train Loss:", float(loss))	
       loss.backward()
       optimizer.step()
       valid_loss = 0