search

leoluopy / GroupFacePytorch

none official pytorch implement of CVPR2020 GroupFace
24 stars 1 forks source link

Bug while training on demo_ims #2

Open rlrahulkanojia opened 3 years ago

rlrahulkanojia commented 3 years ago

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]))

leoluopy commented 3 years ago

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]))