# Trainer API (wandb) error

**URL:** <https://discuss.huggingface.co/t/trainer-api-wandb-error/127342>\
**Category:** Beginners\
**Created:** [November 24, 2024, 1:17pm UTC](https://discuss.huggingface.co/t/trainer-api-wandb-error/127342 "2024-11-24T13:17:20Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![khizarAI](https://avatars.discourse-cdn.com/v4/letter/k/cc9497/32.png) [@khizarAI](https://discuss.huggingface.co/u/khizarAI)\
**Post date:** [November 24, 2024, 1:17pm UTC](https://discuss.huggingface.co/t/trainer-api-wandb-error/127342/1 "2024-11-24T13:17:20Z")

</div>

from transformers import TrainingArguments

batch\_size = 8  
gradient\_accumulation\_steps = 1  
num\_train\_epochs = 10

training\_args = TrainingArguments(  
output\_dir=“DistilHuBert-finetuned-gtzan”,  
eval\_strategy=“epoch”,  
save\_strategy=“epoch”,  
learning\_rate=5e-5,  
per\_device\_train\_batch\_size=batch\_size,  
gradient\_accumulation\_steps=gradient\_accumulation\_steps,  
per\_device\_eval\_batch\_size=batch\_size,  
num\_train\_epochs=num\_train\_epochs,  
warmup\_ratio=0.1,  
logging\_steps=5,  
load\_best\_model\_at\_end=True,  
metric\_for\_best\_model=“accuracy”,  
fp16=True,  
push\_to\_hub=False,  
)

 ![image](https://us1.discourse-cdn.com/hellohellohello/original/3X/f/0/f007bce772133e8f22fa1d5b099ee0128baa665b.png)