search

hunkim / PyTorchZeroToAll

Simple PyTorch Tutorials Zero to ALL!
http://bit.ly/PyTorchZeroAll
3.89k stars 1.2k forks source link

Fix the number of neurons in last fully connected layer #44

Open kHarshit opened 5 years ago

kHarshit commented 5 years ago

Fix number of neurons in last fully connected layer to 2 (equal to number of classes in diabetes dataset).