search

UCSD-E4E / acoustic-multiclass-training

Data processing and training pipeline for classifying bird species by sound
GNU General Public License v3.0
9 stars 4 forks source link

Fix one-hot encoding code in util.py #170

Open TQZhang04 opened 3 weeks ago

TQZhang04 commented 3 weeks ago

Temporary fix to RunTime Error ... detected tensor([0, 1, 2]) expected [0, 1] add torch.clamp(labels_for_score, max=1) after line 72 in train.py

TQZhang04 commented 2 weeks ago

Temporary fix to RunTime Error ... detected tensor([0, 1, 2]) expected [0, 1]: add labels_for_score = torch.clamp(labels_for_score, max=1) after line 72 in train.py