Commit c6a914d0 authored by Lukas Wolf's avatar Lukas Wolf
Browse files

removed debug prints

parent 5cdc8474
......@@ -15,10 +15,8 @@ def train_loop(dataloader, model, loss_fn, optimizer):
Using the given loss_fn and optimizer
Returns training loss of the epoch to be tracked by the caller
"""
#print("Enter training:")
#get_gpu_memory()
#print(torch.cuda.memory_summary())
size = len(dataloader.dataset)
training_loss, correct = 0, 0
for batch, (X, y) in enumerate(dataloader):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment