pytorch-tutorial
pytorch-tutorial copied to clipboard
代码里展示的是 Softmax Regression,而非 Logistic Regression
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
看来不是我一个人有疑问