From 816f8565ca1d21684111b7d72dc7e481f52d92dc Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Wed, 3 May 2023 09:10:26 +0200
Subject: [PATCH] tokenizer were not provided in dataloader, now they are
---
train_lit4rsvqa.py | 1 +
train_rsvqa.py | 1 +
2 files changed, 2 insertions(+)
diff --git a/train_lit4rsvqa.py b/train_lit4rsvqa.py
index bce447f..0e71df0 100644
--- a/train_lit4rsvqa.py
+++ b/train_lit4rsvqa.py
@@ -316,6 +316,7 @@ def main(
num_workers_dataloader=num_workers_dataloader,
batch_size=batch_size,
max_img_idx=max_img_index,
+ tokenizer=hf_tokenizer
)
trainer.fit(model=model, datamodule=dm)
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 4a02025..2286005 100644
--- a/train_rsvqa.py
+++ b/train_rsvqa.py
@@ -339,6 +339,7 @@ def main(
num_workers_dataloader=num_workers_dataloader,
batch_size=batch_size,
max_img_idx=max_img_index,
+ tokenizer=hf_tokenizer
)
trainer.fit(model=model, datamodule=dm)
--
GitLab