Open icemiliang opened 6 years ago
In 02 Logistic Regression, all accuracies are zero. Need to convert torch tensor to float.
line 68: running_acc += num_correct.data[0] --> running_acc += num_correct.data[0].float() line 98: eval_acc += num_correct.data[0] --> eval_acc += num_correct.data[0].float()
In 02 Logistic Regression, all accuracies are zero. Need to convert torch tensor to float.
line 68: running_acc += num_correct.data[0] --> running_acc += num_correct.data[0].float() line 98: eval_acc += num_correct.data[0] --> eval_acc += num_correct.data[0].float()