feat: automatic trigger v2

2023-05-07 19:58:54 +02:00 · 2023-05-07 19:58:54 +02:00 · 8711c17ee8
commit 8711c17ee8
parent 1218d72179
1 changed files with 3 additions and 1 deletions
--- a/2
+++ b/2
@ -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}" 
            }