feat: automatic trigger v2
This commit is contained in:
parent
1218d72179
commit
8711c17ee8
@ -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}"
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user