From d1ba11da5987e933db1c2299900afa2a455de6f9 Mon Sep 17 00:00:00 2001 From: Baris Can Durak Date: Mon, 7 Apr 2025 15:36:20 +0200 Subject: [PATCH] template huggingface fixes --- .github/workflows/ci.yml | 2 +- template/steps/training/model_trainer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index e6a4ffb..779a4e3 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -56,5 +56,5 @@ jobs: with: stack-name: ${{ matrix.stack-name }} python-version: ${{ matrix.python-version }} - ref-zenml: ${{ inputs.ref-zenml || 'develop' }} + ref-zenml: ${{ inputs.ref-zenml || 'bugfix/cleaning-up-the-ci-again' }} ref-template: ${{ inputs.ref-template || github.ref }} diff --git a/template/steps/training/model_trainer.py b/template/steps/training/model_trainer.py index bd5d85b..744c8ec 100644 --- a/template/steps/training/model_trainer.py +++ b/template/steps/training/model_trainer.py @@ -101,7 +101,7 @@ def model_trainer( per_device_eval_batch_size=eval_batch_size, num_train_epochs=num_epochs, weight_decay=weight_decay, - evaluation_strategy='steps', + eval_strategy='steps', save_strategy='steps', save_steps=1000, eval_steps=100,