feat: automatic trigger v2
This commit is contained in:
parent
8711c17ee8
commit
7df11da81d
@ -42,6 +42,7 @@ node {
|
||||
|
||||
sh "kaggle datasets download -p data --unzip clmentbisaillon/fake-and-real-news-dataset && python ./dataset.py --dataset"
|
||||
sh "cp ./data/dataset/* ${WORKSPACE}"
|
||||
sh "cp ./data/*.csv ${WORKSPACE}"
|
||||
|
||||
}
|
||||
}
|
||||
|
@ -50,20 +50,22 @@ node {
|
||||
def dockerImage = docker.build("s424714-model")
|
||||
|
||||
dockerImage.inside {
|
||||
stage("Docker: cloning artifacts"){
|
||||
sh 'mkdir -p ./data/dataset'
|
||||
sh 'mv -t ./data/dataset train.csv test.csv val.csv'
|
||||
}
|
||||
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}"
|
||||
}
|
||||
withEnv(["TRANSFORMERS_CACHE=./.cache"]) {
|
||||
stage("Docker: cloning artifacts"){
|
||||
sh 'mkdir -p ./data/dataset'
|
||||
sh 'mv -t ./data/dataset train.csv test.csv val.csv'
|
||||
sh 'mv -t ./data True.csv Fake.csv'
|
||||
}
|
||||
stage("Docker: Running training model")
|
||||
{
|
||||
sh 'mkdir -p ./.cache'
|
||||
// sh ""
|
||||
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