diff --git a/Jenkinsfile b/Jenkinsfile index 9af78a62..f6168e63 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -22,7 +22,7 @@ pipeline { agent { docker { image 'fnlp:torch-1.11' - args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all --shm-size 256M' + args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all --shm-size 1G' } } steps { @@ -55,19 +55,19 @@ pipeline { sh 'FASTNLP_BACKEND=paddle pytest ./tests/core/controllers/test_trainer_paddle.py --durations=0 --co' } } - stage('Test Jittor') { - agent { - docker { - image 'fnlp:jittor' - args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all' - } - } - steps { - // sh 'pip install fitlog' - // sh 'pytest ./tests --html=test_results.html --self-contained-html' - sh 'pytest ./tests --durations=0 -m jittor --co' - } - } + // stage('Test Jittor') { + // agent { + // docker { + // image 'fnlp:jittor' + // args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all' + // } + // } + // steps { + // // sh 'pip install fitlog' + // // sh 'pytest ./tests --html=test_results.html --self-contained-html' + // sh 'pytest ./tests --durations=0 -m jittor --co' + // } + // } } } } diff --git a/tests/core/callbacks/test_load_best_model_callback_torch.py b/tests/core/callbacks/test_load_best_model_callback_torch.py index 9f346003..c2be7f7a 100644 --- a/tests/core/callbacks/test_load_best_model_callback_torch.py +++ b/tests/core/callbacks/test_load_best_model_callback_torch.py @@ -83,7 +83,6 @@ def test_load_best_model_callback( ): for save_folder in ['save_models', None]: for only_state_dict in [True, False]: - logger.error(f"{save_folder}, {only_state_dict}") callbacks = [LoadBestModelCallback(monitor='acc', only_state_dict=only_state_dict, save_folder=save_folder)] trainer = Trainer(