flip order for convenience

ZIB-IOL · Oct 19, 2023 · 42a6bb2 · 42a6bb2
1 parent 9e5fdbe
commit 42a6bb2
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/src/fw_algorithms.jl b/src/fw_algorithms.jl
@@ -68,12 +68,6 @@ function frank_wolfe(
         @warn("Momentum-averaged gradients should usually be used with agnostic stepsize rules.",)
     end
 
-    # instanciating container for gradient
-    if gradient === nothing
-        gradient = collect(x)
-    end
-
-
     if verbose
         println("\nVanilla Frank-Wolfe Algorithm.")
         NumType = eltype(x0)
@@ -95,18 +89,21 @@ function frank_wolfe(
             x = copyto!(similar(x), x)
         end
     end
+
+    # instanciating container for gradient
+    if gradient === nothing
+        gradient = collect(x)
+    end
+
     first_iter = true
     if linesearch_workspace === nothing
         linesearch_workspace = build_linesearch_workspace(line_search, x, gradient)
     end
 
     # container for direction
     d = similar(x)
-    gtemp = if momentum === nothing
-        d
-    else
-        similar(x)
-    end
+    gtemp = momentum === nothing ? d : similar(x)
+
     while t <= max_iteration && dual_gap >= max(epsilon, eps(float(typeof(dual_gap))))
 
         #####################