From b1084a98fc2e18cc7458fb3ba0ed4f9ca73f2a2a Mon Sep 17 00:00:00 2001 From: "Sam.Porter" Date: Mon, 30 Sep 2024 20:47:24 +0100 Subject: [PATCH] reset initial step size after armijo step --- docker.sh | 16 ++++++++++++++++ main.py | 1 + 2 files changed, 17 insertions(+) create mode 100644 docker.sh diff --git a/docker.sh b/docker.sh new file mode 100644 index 0000000..3c2eefc --- /dev/null +++ b/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 +tensorboard --bind_all --port 6006 --logdir ./output \ No newline at end of file diff --git a/main.py b/main.py index 4039afd..7ab8bdc 100644 --- a/main.py +++ b/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