RitaRamo / smallcap

SmallCap: Lightweight Image Captioning Prompted with Retrieval Augmentation
94 stars 20 forks source link

Possible errors in the source code /file: train.py; function: main #18

Open kaseketsu opened 3 months ago

kaseketsu commented 3 months ago

As shown in the bold part of the following code: trainer = Seq2SeqTrainer( model=model, args=training_args, data_collator=default_data_collator, train_dataset=train_dataset, tokenizer=feature_extractor, )

trainer.train() Perhaps it should be replaced by tokenizer=tokenizer