Closed monajalal closed 3 years ago
[Not sure if 100% makes sense or correct] This change (## refers to old code)
# measure accuracy and record loss
##losses.update(loss_triplet.data[0], num_items)
##accs.update(acc.data[0], num_items)
##emb_norms.update(loss_embed.data[0])
##mask_norms.update(loss_mask.data[0])
losses.update(loss_triplet.data, num_items)
accs.update(acc.data, num_items)
emb_norms.update(loss_embed.data)
mask_norms.update(loss_mask.data)
based on this suggestion https://github.com/NVIDIA/flownet2-pytorch/issues/113#issuecomment-450802359 makes any sense if the error i was getting is non-existent?