Problem with triplet loss

KaiyangZhou / deep-person-reid

Torchreid: Deep learning person re-identification in PyTorch.

https://kaiyangzhou.github.io/deep-person-reid/

MIT License

4.26k stars 1.14k forks source link

Problem with triplet loss #531

Open shayan-aqabarary opened 1 year ago

shayan-aqabarary commented 1 year ago

I have 2 changes in original code, first change network from "resnet" to "osnet" (osnet_x1_0). second change loss function from "softmax" to "triplet". and I get this error. can anyone help me? @KaiyangZhou Screenshot (8)

yajiedesign commented 7 months ago

use ImageTripletEngine