Open icemiliang opened 6 years ago
Accuracies are all zeros. 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()
Accuracies are all zeros. 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()