05_Pytorch going Modular model_builder.py

[kundetiA]

When we create an instance of TinyVgg Model, in arguments we should mention output_shape as len(class_names), but got an error, class_names is not defined in model_builder.py

[pritesh2000]

You have to define class_names before creating instance of TinyVGG

class_names = ['pizza', 'steak', 'sushi']