05_Pytorch going Modular model_builder.py

mrdbourke / pytorch-deep-learning

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

https://learnpytorch.io

MIT License

9.34k stars 2.78k forks source link

05_Pytorch going Modular model_builder.py #884

Open kundetiA opened 3 months ago

kundetiA commented 3 months ago

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 commented 1 week ago

You have to define class_names before creating instance of TinyVGG

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