feat: automatic trigger v2
This commit is contained in:
parent
1218d72179
commit
8711c17ee8
@ -45,7 +45,7 @@ node {
|
|||||||
|
|
||||||
|
|
||||||
sh "chmod +x -R ${env.WORKSPACE}"
|
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")
|
def dockerImage = docker.build("s424714-model")
|
||||||
|
|
||||||
@ -57,6 +57,8 @@ node {
|
|||||||
stage("Docker: Running training model")
|
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 'python ./src/main.py --train --lr=$LR --batch=$BATCH_SIZE --epochs=$NUM_EPOCHS'
|
||||||
sh "cp ./results/model.pt ${WORKSPACE}"
|
sh "cp ./results/model.pt ${WORKSPACE}"
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user