GilesStrong / tomopt

TomOpt: Differential Muon Tomography Optimisation
GNU Affero General Public License v3.0
4 stars 0 forks source link

Refactor warmup cbs #147

Closed GilesStrong closed 1 year ago

GilesStrong commented 1 year ago

Closes #142 closes #143 CostCoefWarmup and PanelOptConfig now both inherit from WarmupCallback and will wait to act in the order that they are provided in the list of callbacks. During warmup, AbsVolumeWrapper will skip the optimiser step call, preventing them from recording gradients or updating the panels. Once all warmup callbacks have finished, the optimiser steps will be called as normal. Additionally, fixes a bug where PanelOptConfig used the absolute value median of the gradients, rather than the median of the absolute values of the gradients.