s424714/ium_424714
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: automatic trigger v2

Browse Source
This commit is contained in:
Filip Patyk 2023-05-07 19:58:54 +02:00
parent 1218d72179
commit 8711c17ee8
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Jenkinsfile-training
View File

@ -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}"
}