diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py index 8408d230..45a00ec0 100644 --- a/docs/source/examples/example_model.py +++ b/docs/source/examples/example_model.py @@ -70,7 +70,6 @@ class ExampleModel(RootModule): # calculate loss loss_val = self.loss(y, y_hat) - print(loss_val) # tqdm_dic = {'tng_loss': loss_val.item()} # return loss_val, tqdm_dic @@ -91,7 +90,6 @@ class ExampleModel(RootModule): y_hat = self.forward(x) loss_val = self.loss(y, y_hat) - print(loss_val) # acc labels_hat = torch.argmax(y_hat, dim=1)