Added GPU use to the compose examples

Author: lapplislazuli

docker-compose-minimal.yml

@@ -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

@@ -21,3 +21,8 @@ services:
       pretrained_model: microsoft/codebert-base
       load_existing_model: "true"
       load_model_path: /models/best_pytorch_model.bin
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]

docker-compose.yml

@@ -16,3 +16,9 @@ services:
       valid_file: /dataset/valid.jsonl
       test_file: /dataset/test.jsonl
       batch_size: 8
+      
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]