Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fixed indentation in the MBCC Jenkins file and also separated out tra…
Browse files Browse the repository at this point in the history
…ining and inference into two separate stages
  • Loading branch information
piyushghai committed Jul 26, 2018
1 parent edd6816 commit 87103d4
Showing 1 changed file with 18 additions and 14 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -49,21 +49,25 @@ def docker_run(platform, function_name, use_nvidia, shared_mem = '500m') {
}

try {
stage('MBCC'){
parallel 'ModelBackwardsCompat: CPU': {
node('mxnetlinux-cpu') {
ws('workspace/modelBackwardsCompat') {

init_git()
// Train models on older versions
docker_run('ubuntu_nightly_cpu', 'nightly_model_backwards_compat_train', false)
// upload files to S3 here outside of the docker environment
sh /tests/nightly/model_backwards_compatibility_check/upload_models_to_s3.sh
// Perform inference on these models
docker_run('ubuntu_nightly_cpu', 'nightly_model_backwards_compat_test', false)
stage('MBCC Train'){
node('mxnetlinux-cpu') {
ws('workspace/modelBackwardsCompat') {
init_git()
// Train models on older versions
docker_run('ubuntu_nightly_cpu', 'nightly_model_backwards_compat_train', false)
// upload files to S3 here outside of the docker environment
sh ./tests/nightly/model_backwards_compatibility_check/upload_models_to_s3.sh
}
}
}

}
}
stage('MBCC Inference)'{
node('mxnetlinux-cpu') {
ws('workspace/modelBackwardsCompat') {
init_git()
// Perform inference on the latest version of MXNet
docker_run('ubuntu_nightly_cpu', 'nightly_model_backwards_compat_test', false)
}
}
}
} catch (caughtError) {
Expand Down

0 comments on commit 87103d4

Please sign in to comment.