reset initial step size after armijo step

docker.sh

@@ -0,0 +1,16 @@
+# 1. git clone & cd to your submission repository
+# 2. mount `.` to container `/workdir`:
+docker run --rm -it --gpus all -p 6006:6006 \
+  -v /path/to/data:/mnt/share/petric:ro \
+  -v .:/workdir -w /workdir synerbi/sirf:edge-gpu /bin/bash
+# 3. install metrics & GPU libraries
+conda install monai tensorboard tensorboardx jupytext cudatoolkit=11.8
+pip uninstall torch # monai installs pytorch (CPU), so remove it
+pip install tensorflow[and-cuda]==2.14  # last to support cu118
+pip install torch --index-url https://download.pytorch.org/whl/cu118
+pip install git+https://github.com/TomographicImaging/Hackathon-000-Stochastic-QualityMetrics
+# 4. optionally, conda/pip/apt install environment.yml/requirements.txt/apt.txt
+# 5. run your submission
+python petric.py &
+# 6. optionally, serve logs at <http://localhost:6006>
+tensorboard --bind_all --port 6006 --logdir ./output

main.py

@@ -148,6 +148,7 @@ def get_step_size(self, algorithm):
 
             # Update the internal state with the new step size as the minimum of the current and previous step sizes
             self.step_size = min(step_size, self.step_size)
+            self.initial_step_size = self.step_size
 
             if self.counter < self.steps:
                 self.counter += 1