How to train on my own dataset?
Open
- Dominant language
- Python
- Stars
- 592
- Forks
- 156
- PR merge metrics
- No merged PRs in 30d
Description
The dataset returns a list of str, so error happens at train_model.py line 79-line81
"
labels = Variable(torch.from_numpy(np.array(labels)).cuda())
labels_swap = Variable(torch.from_numpy(np.array(labels_swap)).cuda())
swap_law = Variable(torch.from_numpy(np.array(swap_law)).float().cuda())
"
cannot convert labels to np.ndarray
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.