To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

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):
......@@ -89,4 +87,4 @@ def validation_loop(dataloader, model, loss_fn):
logging.info(f"Avg validation accuracy {accuracy:>8f}")
return float(loss), float(accuracy)
return float(loss), -1 # Can be used for early stopping
\ No newline at end of file
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