change batch size to 12

Dockerfile

@@ -0,0 +1,7 @@
+# docker build -t v5 .
+FROM registry.webis.de/code-research/tira/tira-user-team-chenteam-chen123hhh/tira:v4-tira-docker-software-id-resonnt-halite
+
+ADD test.py /workspace/clef/test.py
+
+ENTRYPOINT [ "python", "/workspace/clef/test.py", "$inputDataset/dataset.jsonl", "$outputDir" ]
+

README.md

@@ -0,0 +1,18 @@
+
+Build via:
+
+```
+docker build -t v5 .
+```
+
+Run via:
+
+```
+tira-run \
+  --input-dataset generative-ai-authorship-verification-panclef-2024/pan24-generative-authorship-tiny-smoke-20240417-training \
+  --image v5 \
+  --tira-vm-id team-chenteam-chen123hhh \
+  --skip-local-test \
+  --push true
+```
+

test.py

@@ -81,8 +81,8 @@ def __getitem__(self, idx):
     # os.environ["CUDA_VISIBLE_DEVICES"] = '2'
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
     print(f'Using {device} device')
-    batch_size = 24
-    checkpoint = "/t-ng/Bert/bert_model/MyModel"
+    batch_size = 8
+    checkpoint = "/workspace/clef/MyModel"
     input_path = sys.argv[1]
     output_dir = sys.argv[2]
     # --------------------------------------------------------------------------