-
Notifications
You must be signed in to change notification settings - Fork 75
Open
Description
hi @anewell ,
Thank you for releasing the training code.
I tried to train the network on coco2014 with original settings except {'batchsize': 8, 'input_res': 256, 'output_res': 64}. The training loss curve looks so weird but the output map look right. I'm quite confused. Does this normal? Could you please share your training log?
In the line 102 of my_lib.c for push_loss calculation, why do you need to multiply the push_loss with 0.5?
-
if(current_people>1)
-
output_tmp[0]/= current_people*(current_people-1)/2;
-
output_tmp[0] *= 0.5;
Why are the weights for pull_loss and push_loss so small(1e-3)? How did you choose the weights for pull_loss, push_loss and detection_loss?
Metadata
Metadata
Assignees
Labels
No labels