Merge pull request #13 from milzj/devel

Devel

Author: milzj

.github/workflows/test-FW4PDE.yml

@@ -16,9 +16,9 @@ jobs:
       run:
         shell: bash -l {0}
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - name: Setup Miniconda
-        uses: conda-incubator/setup-miniconda@v2.2.0
+        uses: conda-incubator/setup-miniconda@v3
         with:
           activate-environment: anaconda-client-env
           environment-file: environment.yml

examples/convex/example57/example57.py

@@ -74,6 +74,11 @@
     plot(solution_final)
     plt.savefig("solution.pdf")
 
+    gradient_final = sol["gradient_final"].data
+    c = plot(gradient_final)
+    plt.colorbar(c)
+    plt.savefig("gradient_final.pdf")
+
     solution_final = sol["control_final"]
     obj = problem.obj
     obj(solution_final)

fw4pde/algorithms/__init__.py

@@ -1,3 +1,2 @@
 from .box_lmo import NumpyBoxLMO, MoolaBoxLMO
 from .frank_wolfe import FrankWolfe
-from .frank_wolfe_memory import FrankWolfeMemory

fw4pde/algorithms/frank_wolfe.py

@@ -160,6 +160,7 @@ def solve(self):
 
             # Update stats
             self.data.update({"control_final": u})
+            self.data.update({"gradient_final": gradient})
             objective_final = obj_u + nonsmooth_obj(u.data)
             self.data.update({"objective_final": objective_final})
 
@@ -169,6 +170,7 @@ def solve(self):
             if objective_final < objective_best:
 
                 self.data.update({"control_best": u})
+                self.data.update({"gradient_best": gradient})
                 self.data.update({"objective_best": objective_final})
                 objective_best = objective_final