Updated Compose-Versions

Author: lapplislazuli

docker-compose-minimal.yml

@@ -1,4 +1,4 @@
-version: '3.3'
+version: '3.8'
 
 services:
   experiment:
@@ -10,4 +10,9 @@ services:
       epochs: 10
       train_file: /dataset/train_minimal.jsonl
       valid_file: /dataset/valid_minimal.jsonl
-      test_file: /dataset/test_minimal.jsonl
+      test_file: /dataset/test_minimal.jsonl
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]

docker-compose-pretrained-minimal.yml

@@ -1,12 +1,8 @@
-version: '3.3'
+version: '3.8'
 
 services:
   experiment:
     image: ciselab/codebert-code2text
-    cpu_count: 8
-    mem_limit: 32g
-    oom_kill_disable: true
-    oom_score_adj: 500
     volumes:
       - ./dataset/java/:/dataset:ro
       - ./compose_output:/experiment/output

docker-compose-python.yml

@@ -1,4 +1,4 @@
-version: '3.3'
+version: '3.8'
 
 services:
   python-codebert-training:

docker-compose.yml

@@ -1,12 +1,8 @@
-version: '3.3'
+version: '3.8'
 
 services:
   experiment:
     image: ciselab/codebert-code2text
-    cpu_count: 8
-    mem_limit: 32g
-    oom_kill_disable: true
-    oom_score_adj: 500
     volumes:
       - ./dataset/java/:/dataset:ro
       - ./compose_output:/experiment/output
@@ -15,7 +11,7 @@ services:
       train_file: /dataset/train.jsonl
       valid_file: /dataset/valid.jsonl
       test_file: /dataset/test.jsonl
-      batch_size: 32
+      batch_size: 8
     deploy:
       resources:
         reservations: