Docker files updated

Author: gferrate

Makefile

@@ -11,8 +11,7 @@ update:
 	docker-compose down
 	docker build -t pytorch_fl .
 	docker-compose up -d
-prepare:
-	touch logs/client_8003.log
-	touch logs/client_8004.log
-	touch logs/main_server.log
-	touch logs/secure_aggregator.log
+
+del_logs:
+	tar -czvf logs_bak.tar.gz logs/*.log
+	rm logs/*.log

docker-compose.yml

@@ -22,6 +22,7 @@ services:
             - PORT=8001
         volumes:
             - ./logs:/code/logs:rw
+            - ./data/classification:/code/data/classification:rw
         depends_on:
             - main_server
 
@@ -36,6 +37,7 @@ services:
             - N_CLIENT=0
         volumes:
             - ./logs:/code/logs:rw
+            - ./data/classification:/code/data/classification:rw
         depends_on:
             - secure_aggregator
 
@@ -50,6 +52,7 @@ services:
             - N_CLIENT=1
         volumes:
             - ./logs:/code/logs:rw
+            - ./data/classification:/code/data/classification:rw
         depends_on:
             - secure_aggregator
 

docker-entrypoint.sh

@@ -2,14 +2,13 @@
 
 if [ "$TYPE" = "CLIENT" ]; then
     echo "Starting client"
-    echo /code/logs/client_$PORT.log
-    python client/app.py -p $PORT -n $N_CLIENT &> /code/logs/client_$PORT.log
+    python client/app.py -p $PORT -n $N_CLIENT 
 elif [[ "$TYPE" = "MAIN_SERVER" ]]; then
     echo "Starting main server"
-    python main_server/app.py -p $PORT &> /code/logs/main_server.log
+    python main_server/app.py -p $PORT 
 elif [[ "$TYPE" = "SECURE_AGGREGATOR" ]]; then
     echo "Starting secure aggregator"
-    python secure_aggregator/app.py -p $PORT &> /code/logs/secure_aggregator.log
+    python secure_aggregator/app.py -p $PORT 
 else
     echo "Type not recognized"
 fi