https://github.com/KMnP/fashionpedia-api | |
Trained "detection-datasets/fashionpedia" hugging face dataset with the following params: | |
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, | |
) |