refactored model tests

2026-09-11 12:31:23 +08:00 · 2019-07-24 13:46:32 -04:00
parent 3d31219c85
commit aba7006fc2
1 changed files with 2 additions and 0 deletions
@@ -98,8 +98,10 @@ class LightningTemplateModel(LightningModule):
        """
        x, y = data_batch
        x = x.view(x.size(0), -1)
+        print('-'*100)
        print('x: ', x.device)
        print('model: ', self.c_d1.weight.device)
+        print('-'*100)
        y_hat = self.forward(x)

        loss_val = self.loss(y, y_hat)