Issue in 02_CustomScript_Training_Pipeline

[bhavangowdan]

Hi , while running the cell 22 in 6.4 Visualize Performance across models of the notebook 02_CustomScript_Training_Pipeline.ipynb we are getting the error mentioned below


Could you please check into this?