yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

y1x4 opened this issue · comments

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

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

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

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

看来不是我一个人有疑问