Self-restrained-Triplet-Loss
Self-restrained-Triplet-Loss copied to clipboard
This line should be corrected.
https://github.com/fdbtrs/Self-restrained-Triplet-Loss/blob/adb9421fb8171b530d1351a59763c3be6f0fd346/main.py#L141
It should be
cnn_optimizer.zero_grad()