From 6ea968d4b29af3ecb88d97305f62f3218ee1c0e5 Mon Sep 17 00:00:00 2001
From: friebolin <friebolin@cl.uni-heidelberg.de>
Date: Fri, 24 Feb 2023 15:46:43 +0100
Subject: [PATCH] Fix dataloader input
---
inference.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/inference.py b/inference.py
index 313bfad..9785f35 100644
--- a/inference.py
+++ b/inference.py
@@ -81,11 +81,11 @@ model = torch.load(model_path, map_location=device)
model.eval()
-train_sampler = RandomSampler(data_sample)
-train_dataloader = DataLoader(data_sample, sampler=train_sampler, batch_size=1)
+train_sampler = RandomSampler(input_as_dataset)
+train_dataloader = DataLoader(input_as_dataset, sampler=train_sampler, batch_size=1)
for batch in train_dataloader:
- print(batch)
+ #print(batch)
inputs = {'input_ids': batch[0],
'attention_mask': batch[1],
'token_type_ids': batch[2],
--
GitLab