hfl

intel · Feb 14, 2022 · 5f807e0 · 5f807e0
1 parent cf42740
commit 5f807e0
Show file tree

Hide file tree

Showing 6 changed files with 1 addition and 37 deletions.
diff --git a/cczoo/horizontal_fl/hfl-tensorflow/__pycache__/cifar10_input.cpython-36.pyc b/cczoo/horizontal_fl/hfl-tensorflow/__pycache__/cifar10_input.cpython-36.pyc
diff --git a/cczoo/horizontal_fl/hfl-tensorflow/__pycache__/resnet_model.cpython-36.pyc b/cczoo/horizontal_fl/hfl-tensorflow/__pycache__/resnet_model.cpython-36.pyc
diff --git a/cczoo/horizontal_fl/hfl-tensorflow/ps0-python.log b/cczoo/horizontal_fl/hfl-tensorflow/ps0-python.log
diff --git a/cczoo/horizontal_fl/hfl-tensorflow/test-sgx.sh b/cczoo/horizontal_fl/hfl-tensorflow/test-sgx.sh
@@ -50,22 +50,15 @@ function make_custom_env() {
 ROLE=$1
 PS_HOSTS=$2
 WORKER_HOSTS=$3
+
 if [ "$ROLE" == "make" ]; then
-    # rm -rf model *.log
     make clean && make | make_logfilter
 elif [ "$ROLE" == "ps0" ]; then
     make_custom_env
     taskset -c 0-3 stdbuf -o0 gramine-sgx python -u train.py --task_index=0 --job_name=ps $PS_HOSTS $WORKER_HOSTS 2>&1 | runtime_logfilter | tee -a ps0-gramine-python.log &
     if [ "$DEBUG" != "0" ]; then
         wait && kill -9 `pgrep -f gramine`
     fi
-elif [ "$ROLE" == "ps1" ]; then
-    make_custom_env
-    taskset -c 4-7 stdbuf -o0 gramine-sgx python -u train.py --task_index=1 --job_name=ps $PS_HOSTS $WORKER_HOSTS 2>&1  | runtime_logfilter | tee -a ps1-gramine-python.log &
-    if [ "$DEBUG" != "0" ]; then
-        wait && kill -9 `pgrep -f gramine`
-    fi
-
 elif [ "$ROLE" == "worker0" ]; then
     make_custom_env
     taskset -c 8-11 stdbuf -o0 gramine-sgx python -u train.py --task_index=0 --job_name=worker $PS_HOSTS $WORKER_HOSTS 2>&1 | runtime_logfilter | tee -a worker0-gramine-python.log &

diff --git a/cczoo/horizontal_fl/hfl-tensorflow/test.sh b/cczoo/horizontal_fl/hfl-tensorflow/test.sh
@@ -39,9 +39,6 @@ WORKER_HOSTS=$3
 if [ "$ROLE" == "ps0" ]; then
     make_custom_env
     taskset -c 0-3 stdbuf -o0 python -u train.py --task_index=0 --job_name=ps 2>&1 $PS_HOSTS $WORKER_HOSTS | tee -a ps0-python.log &
-elif [ "$ROLE" == "ps1" ]; then
-    make_custom_env
-    taskset -c 4-7 stdbuf -o0 python -u train.py --task_index=1 --job_name=ps 2>&1 $PS_HOSTS $WORKER_HOSTS | tee -a ps1-python.log &
 elif [ "$ROLE" == "worker0" ]; then
     make_custom_env
     taskset -c 8-11 stdbuf -o0 python -u train.py --task_index=0 --job_name=worker $PS_HOSTS $WORKER_HOSTS 2>&1 | tee -a worker0-python.log &

diff --git a/cczoo/horizontal_fl/hfl-tensorflow/worker0-python.log b/cczoo/horizontal_fl/hfl-tensorflow/worker0-python.log