How to use BCEWithLogitsLossFlat in lesson1-pets.ipynb

MultiCategoryBlock expect a list of labels, so you have to change your label_func slightly to return something like [“cat”] (and not “cat”).

3 Likes