Adds docs

peterdsharpe · Oct 5, 2024 · e287b07 · e287b07
1 parent a964518
commit e287b07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/gen2_architecture/train_blind_neural_network.py b/training/gen2_architecture/train_blind_neural_network.py
@@ -236,7 +236,7 @@ def loss_function(y_pred, y_data, return_individual_loss_components=False):
 
     n_batches_per_epoch = len(train_loader)
 
-    num_epochs = 100000000
+    num_epochs = 10 ** 9  # Effectively loop until manually stopped
     for epoch in range(num_epochs):
         # Put the model in training mode
         net.train()