search

mrdbourke / pytorch-deep-learning

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.
https://learnpytorch.io
MIT License
10.49k stars 3.09k forks source link

Update 03_pytorch_computer_vision_exercise_solutions.ipynb #938

Closed Hrut7hesh closed 1 month ago

Hrut7hesh commented 4 months ago

This update modifies the initialization of the Accuracy function to explicitly specify the task parameter as 'multiclass'. This change is necessary due to updates in the TorchMetrics library, which now requires the task parameter to be specified when creating an instance of the Accuracy class.

mrdbourke commented 1 month ago

Thank you!