wandb에서 epoch 기준으로 log 찍는 법

[sne12345]

1. train.py 에서 wandb.log에 epoch 추가하기

if args.wandb == True:
wandb.log({
"Train/Train loss": round(loss.item(), 4),
"Train/Train mIoU": round(mIoU.item(), 4),
"Train/Train acc": round(acc.item(), 4),
"learning_rate": current_lr,
"epoch":epoch+1
})

if args.wandb == True:
wandb.log({
"Media/predict images": figure,
"Valid/Valid loss": round(avg_loss.item(), 4),
"Valid/Valid mIoU": round(mIoU, 4),
"Valid/Valid acc": round(acc, 4),
"Metric/Background_IoU": IoU_by_class[0]['Background'], "Metric/General_trash_IoU": IoU_by_class[1]['General trash'], "Metric/Paper_IoU": IoU_by_class[2]['Paper'],
"Metric/Paper_pack_IoU": IoU_by_class[3]['Paper pack'], "Metric/Metal_IoU": IoU_by_class[4]['Metal'], "Metric/Glass_IoU": IoU_by_class[5]['Glass'],
"Metric/Plastic_IoU": IoU_by_class[6]['Plastic'], "Metric/Styrofoam_IoU": IoU_by_class[7]['Styrofoam'], "Metric/Plastic_bag_IoU": IoU_by_class[8]['Plastic bag'],
"Metric/Battery_IoU": IoU_by_class[9]['Battery'], "Metric/Clothing_IoU": IoU_by_class[10]['Clothing'],
"epoch":epoch+1
})

2. wandb에서 X Axis를 epoch로 설정하기

걸리는점

wandb.log에 step을 넣어주면 에러가 나서 뺐는데, 혹시 문제가 있을까요?

[ppskj178]

step은 제가 임시로 넣고 테스트 요청드렸던거 같은데
재환님이 테스트 돌리고 계셔서 잘된줄 알았는데 아니였나요?

근데 epoch 단위로 찍어도 추세를 알 수 있겠으면 상관없을 것 같은데
좀 작지 않나요?

[sne12345]

그러게요 에포크당 하나씩밖에 안 찍어줘서 데이터가 적긴 하네요..훔

[jaehwan-AI]

step으로 잘 돌아갑니다. 다만 log가 누적합으로 찍혀서 조금 수정한 부분은 있습니다.
상황에 맞게 step과 epoch을 사용하면 될 것 같아요!