GilesStrong · GilesStrong · Feb 12, 2024 · Feb 10, 2024 · Feb 10, 2024 · Feb 12, 2024
diff --git a/examples/benchmarks/ladle_furnace.ipynb b/examples/benchmarks/ladle_furnace.ipynb
diff --git a/examples/benchmarks/ladle_furnace_plots/baseline_bias.pdf b/examples/benchmarks/ladle_furnace_plots/baseline_bias.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/baseline_mse.pdf b/examples/benchmarks/ladle_furnace_plots/baseline_mse.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/baseline_predictions.pdf b/examples/benchmarks/ladle_furnace_plots/baseline_predictions.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/initial_bias.pdf b/examples/benchmarks/ladle_furnace_plots/initial_bias.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/initial_mse.pdf b/examples/benchmarks/ladle_furnace_plots/initial_mse.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/initial_predictions.pdf b/examples/benchmarks/ladle_furnace_plots/initial_predictions.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/optimised_2_bias.pdf b/examples/benchmarks/ladle_furnace_plots/optimised_2_bias.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/optimised_2_mse.pdf b/examples/benchmarks/ladle_furnace_plots/optimised_2_mse.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/optimised_2_predictions.pdf b/examples/benchmarks/ladle_furnace_plots/optimised_2_predictions.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/optimised_bias.pdf b/examples/benchmarks/ladle_furnace_plots/optimised_bias.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/optimised_mse.pdf b/examples/benchmarks/ladle_furnace_plots/optimised_mse.pdf
diff --git a/examples/benchmarks/ladle_furnace_plots/optimised_predictions.pdf b/examples/benchmarks/ladle_furnace_plots/optimised_predictions.pdf
diff --git a/examples/getting_started/00_Hello_World.ipynb b/examples/getting_started/00_Hello_World.ipynb
diff --git a/examples/getting_started/01_Indepth_tutorial_single_cycle.ipynb b/examples/getting_started/01_Indepth_tutorial_single_cycle.ipynb
diff --git a/examples/getting_started/02_Indepth_tutorial_optimisation_and_callbacks.ipynb b/examples/getting_started/02_Indepth_tutorial_optimisation_and_callbacks.ipynb
diff --git a/examples/getting_started/03_fixed_budget_mode.ipynb b/examples/getting_started/03_fixed_budget_mode.ipynb
diff --git a/examples/panel_detectors/00_Hello_World.ipynb b/examples/panel_detectors/00_Hello_World.ipynb
diff --git a/examples/panel_detectors/01_Indepth_tutorial_single_cycle.ipynb b/examples/panel_detectors/01_Indepth_tutorial_single_cycle.ipynb
diff --git a/examples/panel_detectors/02_Indepth_tutorial_optimisation_and_callbacks.ipynb b/examples/panel_detectors/02_Indepth_tutorial_optimisation_and_callbacks.ipynb
diff --git a/examples/panel_detectors/03_fixed_budget_mode.ipynb b/examples/panel_detectors/03_fixed_budget_mode.ipynb
diff --git a/examples/panel_detectors/imgs/panel detector.png b/examples/panel_detectors/imgs/panel detector.png
diff --git a/examples/panel_detectors/imgs/volume_layout.png b/examples/panel_detectors/imgs/volume_layout.png
diff --git a/tomopt/benchmarks/ladle_furnace/inference.py b/tomopt/benchmarks/ladle_furnace/inference.py
@@ -11,7 +11,7 @@
 from ...optimisation.callbacks import Callback
 from ...volume import Volume
 
-__all__ = ["EdgeDetLadleFurnaceFillLevelInferrer", "PocaZLadleFurnaceFillLevelInferrer"]
+__all__ = ["EdgeDetLadleFurnaceFillLevelInferrer", "PocaZLadleFurnaceFillLevelInferrer", "LinearCorrectionCallback"]
 
 
 class EdgeDetLadleFurnaceFillLevelInferrer(AbsIntClassifierFromX0):
@@ -416,7 +416,7 @@ def on_x0_pred_end(self) -> None:
 
         self.wrapper.fit_params.pred = self.linear(self.wrapper.fit_params.pred)
 
-    def on_backwards_begin(self) -> None:
+    def on_backwards_end(self) -> None:
         x = torch.stack(self._preds, 0)
         y = torch.stack(self._targs, 0)
         for i in range(10000):

diff --git a/tomopt/optimisation/callbacks/opt_callbacks.py b/tomopt/optimisation/callbacks/opt_callbacks.py
@@ -11,7 +11,7 @@
 if TYPE_CHECKING:
     from ...optimisation.wrapper import AbsVolumeWrapper
 
-__all__ = ["OneCycle"]
+__all__ = ["OneCycle", "EpochSave"]
 
 
 class AbsOptSchedule(PostWarmupCallback, metaclass=ABCMeta):