From e1c458e74d294417a46f243d97a6c371361bf7f6 Mon Sep 17 00:00:00 2001
From: friebolin <friebolin@cl.uni-heidelberg.de>
Date: Fri, 24 Feb 2023 15:33:09 +0100
Subject: [PATCH] Add print
---
inference.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/inference.py b/inference.py
index 6caeb49..2338d33 100644
--- a/inference.py
+++ b/inference.py
@@ -76,7 +76,7 @@ input_as_dataset = Code.preprocess.tokenizer_new(tokenizer, data_sample, max_len
model = Code.models.WordClassificationModel("bert-base-uncased")
-model_path = "saved_models/bert.pth"
+model_path = "Code/saved_models/bert.pth"
model = torch.load(model_path, map_location=device)
model.eval()
@@ -85,6 +85,7 @@ train_sampler = RandomSampler(data_sample)
train_dataloader = DataLoader(data_sample, sampler=train_sampler, batch_size=1)
for batch in train_dataloader:
+ print(batch)
inputs = {'input_ids': batch[0],
'attention_mask': batch[1],
'token_type_ids': batch[2],
--
GitLab