Training Neural Networks in PyTorch

preview_player
Показать описание
Рекомендации по теме
Комментарии
Автор

which video goes into the train/test data loader code?

mitchelob
Автор

How does loss_function compare probabilities (if 10 classes then 10 probabilities) with actual labels (only a single value) while training? model prediction should not be probabilities, they should be actual predicted values. Does it map the highest probability by using max() of the class and output a single value?

vikrampande