GroupFacePytorch icon indicating copy to clipboard operation
GroupFacePytorch copied to clipboard

Bug while training on demo_ims

Open rlrahulkanojia opened this issue 4 years ago • 1 comments

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

rlrahulkanojia avatar Jan 19 '21 05:01 rlrahulkanojia

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

leoluopy avatar Jan 22 '21 01:01 leoluopy