Fashionpedia57000 / README.md
bigmouse's picture
Create README.md
275dee0
|
raw
history blame
473 Bytes
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,
)