Output of PenultimateLayer

🤗Transformers
1

I trained a model on

model = AutoModelForImageClassification.from_pretrained('microsoft/dit-base',num_labels=10)

trainer = Trainer(
    model=model, 
    data_collator=collate_fn,                      
    args=training_args,                  
    train_dataset=prepared_ds['train'],       
    compute_metrics=compute_metrics,
)
trainer.train()

How to I get the output from penultimate layer?