Fine-tuning CV: các lớp mô hình

Trong bài tập này, bạn sẽ tải mô hình đã được pretrained và điều chỉnh đầu ra để phù hợp với một bài toán phân loại mới về các kiểu mẫu xe hơi từ Stanford Cars dataset, thay vì 1000 lớp được dùng cho huấn luyện ImageNet gốc. Bộ dữ liệu này chứa các ảnh ô tô đã được gán nhãn.

Bộ dữ liệu đã được tải (dataset), cũng như AutoModelForImageClassification từ transformers. Bộ dữ liệu đã được lọc để chỉ bao gồm ba kiểu mẫu xe.