GroupFacePytorch
GroupFacePytorch copied to clipboard
Bug while training on demo_ims
for line 77 in arcface_loss.py
generated_len = int(expected_len_centers - positeve_centerIdxs.shape[0])
to
generated_len = abs(int(expected_len_centers - positeve_centerIdxs.shape[0]))
thanks for following , the expected_len_centers should be larger than positeve_centerIdxs.shape[0] . it is used when the center size is too much for training .
for line 77 in arcface_loss.py
generated_len = int(expected_len_centers - positeve_centerIdxs.shape[0])
to
generated_len = abs(int(expected_len_centers - positeve_centerIdxs.shape[0]))