diff --git a/.github/workflows/publish_images.yml b/.github/workflows/publish_images.yml index 63717276..3abaac41 100644 --- a/.github/workflows/publish_images.yml +++ b/.github/workflows/publish_images.yml @@ -88,7 +88,7 @@ jobs: with: context: . push: true - tags: ${{ env.image_name }}:latest + tags: ${{ env.REGISTRY }}/${{ env.image_name }}:latest build-args: | python_version=${{ matrix.python_version }} target=${{ matrix.target }} diff --git a/Dockerfile b/Dockerfile index 0db19548..ab4379bd 100644 --- a/Dockerfile +++ b/Dockerfile @@ -39,7 +39,6 @@ RUN if [ "$target" = "hannah" ] || [ "$target" = "mlonmcu" ]; then\ apt-get update -y && apt-get -y install git mesa-utils libblas-dev liblapack-dev libsndfile1-dev libsox-dev cmake ninja-build curl build-essential; \ fi -RUN echo "Target: $target" # Install poetry using recommended method RUN pip install poetry diff --git a/hannah/train.py b/hannah/train.py index 0d89ad32..3297a443 100644 --- a/hannah/train.py +++ b/hannah/train.py @@ -120,9 +120,6 @@ def train( logger.append(CSVLogger(".", version="logs", name="")) logger.append(JSONLogger(".", version="logs", name="")) - # if DVCLIVE_AVAILABLE: - # logger.append(DVCLogger()) - callbacks = [] callbacks.extend(list(common_callbacks(config)))