Fashionpedia57000

🚀 Fashionpedia API項目

本項目使用特定參數對Hugging Face上的“detection - datasets/fashionpedia”數據集進行訓練，以實現相關的檢測功能，為時尚領域的數據處理和分析提供支持。

🚀 快速開始

訓練參數設置

以下是訓練時使用的參數設置代碼：

training_args = TrainingArguments(
    output_dir="trained_model",
    per_device_train_batch_size=8,
    num_train_epochs=10,
    fp16=False,
    save_steps=200,
    logging_first_step=True,
    logging_steps=10,
    learning_rate=1e-5,
    weight_decay=1e-4,
    save_total_limit=2,
    remove_unused_columns=False,
    push_to_hub=False,
)

代碼解釋

• output_dir：訓練模型的保存目錄，這裡設置為trained_model。
• per_device_train_batch_size：每個設備的訓練批次大小，設置為8。
• num_train_epochs：訓練的輪數，設置為10。
• fp16：是否使用半精度浮點數進行訓練，這裡設置為False。
• save_steps：每訓練200步保存一次模型。
• logging_first_step：是否記錄第一步的訓練日誌，設置為True。
• logging_steps：每10步記錄一次訓練日誌。
• learning_rate：學習率設置為1e - 5。
• weight_decay：權重衰減設置為1e - 4。
• save_total_limit：最多保存2個模型。
• remove_unused_columns：是否移除未使用的列，設置為False。
• push_to_hub：是否將模型推送到Hugging Face Hub，設置為False。