MLOps-Basics icon indicating copy to clipboard operation
MLOps-Basics copied to clipboard

Published 20 hours ago • verified publisher icon graviraja

Key error on Week1

Open treblenalto opened this issue 3 years ago • 2 comments

Hi @graviraja, I was following your tutorial on wandb logging and found a potential error in the training code when visualizing poorly performed data with wandb table. https://github.com/graviraja/MLOps-Basics/blob/13060e0295c7bbc1551e82fe49b6788a5a13ad97/week_1_wandb_logging/train.py#L19-L21 When running, this results in KeyError : "sentence", referring to line 21 sentences = val_batch["sentence"].

I think this is because "sentence" is not part of the columns setup in val_data of data.py. Please correct me if I'm wrong. Thanks :) https://github.com/graviraja/MLOps-Basics/blob/13060e0295c7bbc1551e82fe49b6788a5a13ad97/week_1_wandb_logging/data.py#L29-L42

treblenalto avatar Feb 04 '22 01:02 treblenalto

Hi @Taehee-K , I also got the same error as you. Did you fix that error? Can you show me the results?

PTson2207 avatar Jul 10 '22 06:07 PTson2207