From 8711c17ee81e0134c001ae7d3f48e0d5a9434da3 Mon Sep 17 00:00:00 2001 From: s424714 Date: Sun, 7 May 2023 19:58:54 +0200 Subject: [PATCH] feat: automatic trigger v2 --- Jenkinsfile-training | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/Jenkinsfile-training b/Jenkinsfile-training index fccf588..bf74005 100644 --- a/Jenkinsfile-training +++ b/Jenkinsfile-training @@ -45,7 +45,7 @@ node { sh "chmod +x -R ${env.WORKSPACE}" - copyArtifacts fingerprintArtifacts: true, projectName: 's424714-create-dataset', selector: buildParameter('BUILD_SELECTOR') + copyArtifacts fingerprintArtifacts: true, projectName: 's424714-create-dataset', selector: buildParameter('BUILD_SELECTOR') def dockerImage = docker.build("s424714-model") @@ -57,6 +57,8 @@ node { stage("Docker: Running training model") { + sh 'mkdir -p ./.cache' + sh "export TRANSFORMERS_CACHE=./.cache" sh 'python ./src/main.py --train --lr=$LR --batch=$BATCH_SIZE --epochs=$NUM_EPOCHS' sh "cp ./results/model.pt ${WORKSPACE}" }