From e21589a0d93dfe2637a7b900647397af41ae5bae Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Wed, 18 May 2022 10:03:18 +0000
Subject: [PATCH] =?UTF-8?q?=E6=9A=82=E6=97=B6=E8=B7=B3=E8=BF=87jittor?=
 =?UTF-8?q?=E7=9A=84=E6=B5=8B=E8=AF=95=EF=BC=9B=E5=A2=9E=E5=8A=A0torch1.11?=
 =?UTF-8?q?=E7=9A=84=E5=85=B1=E4=BA=AB=E5=86=85=E5=AD=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Jenkinsfile                                   | 28 +++++++++----------
 .../test_load_best_model_callback_torch.py    |  1 -
 2 files changed, 14 insertions(+), 15 deletions(-)

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(