updated args

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)