Tutorial for multimodal_transformers

train.py

from transformers import Trainer, TrainingArguments

training_args = TrainingArguments(
    output_dir="./logs/model_name",
    logging_dir="./logs/runs",
    overwrite_output_dir=True,
    do_train=True,
    per_device_train_batch_size=32,
    num_train_epochs=1,
    evaluate_during_training=True,
    logging_steps=25,
)

trainer = Trainer(
  model=model,
  args=training_args,
  train_dataset=train_dataset
)

trainer.train()