Open y1x4 opened 5 years ago
pytorch-tutorial/tutorials/01-basics/logistic_regression/main.py
model = nn.Linear(input_size, num_classes) criterion = nn.CrossEntropyLoss()
model = nn.Linear(input_size, num_classes)
criterion = nn.CrossEntropyLoss()
难道不就是用 softmax regression 嘛 (which is used for multi-class classification problems)
是的,没太看明白,注释里写的是Logistic Regression
看来不是我一个人有疑问
pytorch-tutorial/tutorials/01-basics/logistic_regression/main.py
model = nn.Linear(input_size, num_classes)
criterion = nn.CrossEntropyLoss()