Suggestion: Modify the documentation in example notebooks

[Het-Shah]

In the pytorch example notebook, for the annotation format of train_labels.csv file, the name of columns is given as "Id" and "Labels". Upon keeping this as the column names for my CSV file I got an error for wrong column names. Upon looking at sample datasets I found that the actual names for the columns are supposed to be "ID" and "Label". Please rectify to avoid confusion.

[abhi-kumar]

Thank you for pointing it out. Can you please reformat the associated code so that the column names are dynamic and not restricted to "ID and Label" . You may use pandas library to do the same

[Het-Shah]

Ok sure will do it