File size: 473 Bytes
275dee0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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,
) |