Wandb plot x-axis epoch instead of global steps?

[Nevermetyou]

Hi

I am logging my training information in wandb. here is my training args

args = TrainingArguments(
    output_dir="./bert-finetuned-ner",
    overwrite_output_dir=True,
    evaluation_strategy="epoch",
    save_strategy="epoch",
    save_total_limit=5,
    logging_strategy="epoch",
    learning_rate=2e-5,
    num_train_epochs=3,
    weight_decay=0.01,
    report_to="wandb",
    run_name=f"demo_{demo_num}", 
    load_best_model_at_end=True,
)

After the training is done, I go to wandb to inspect the training history. Most of the plot has global_steps as x-axis eventhough I set logging_strategy=“epoch”. Here is my wandb.

All plots in eval tabs has global steps as x-axis as well. I don’t understand this. Can’t we have epoch as x-axis. I try to switch using wandb ui but got an error. Also I don’t understand the meaning of train/epoch plot.

Could someone help me with these, please?

[dblakely]

You should be able to change it by clicking the button on the top right that shows an “x” with an arrow underneath it

[Nevermetyou]

Yes, This seems to work. THANKS!!