max_epochs

3dem · Oct 25, 2023 · 8d324c2 · 8d324c2
1 parent 02f47ba
commit 8d324c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dynamight/deformations/optimize_deformations.py b/dynamight/deformations/optimize_deformations.py
@@ -441,7 +441,7 @@ def optimize_deformations(
             cols = torch.arange(len(particle_dataset))/len(particle_dataset)
 
         # the actual training loop
-        for epoch in range(n_epochs):
+        for epoch in range(2*n_epochs):
             abort_if_relion_abort(output_directory)
             # first, recompute the graphs
             with torch.no_grad():