yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers
MIT License
29.79k stars 8.03k forks source link

代码里展示的是 Softmax Regression,而非 Logistic Regression #192

Open y1x4 opened 4 years ago

y1x4 commented 4 years ago

pytorch-tutorial/tutorials/01-basics/logistic_regression/main.py

model = nn.Linear(input_size, num_classes) criterion = nn.CrossEntropyLoss()

OrangeC93 commented 4 years ago

难道不就是用 softmax regression 嘛 (which is used for multi-class classification problems)

yuzi1949 commented 4 years ago

是的,没太看明白,注释里写的是Logistic Regression

nuaa030710312 commented 2 years ago

看来不是我一个人有疑问