Danlan Chen

margin loss wrong

2

Hello, I think your magin loss is wrong, according to the paper, after relu, there is a square you forgot

test.sh failed

6

Hello, Xinggang, I am getting the following error after I run the test.sh in experiment folder, could you tell me what is wrong here, I googled some result, saying I...

https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/YOLO/train.py change that line to `DEVICE = "cuda" if torch.cuda.is_available() else "cpu"` works on cpu machine

weights path

1

Hello, I am just wondering where are the weights path for resnet 50, it seems that it is deleted.

it is not working with using cuda , it requires using cuda 5.5, however my linux system is 14. which can not install cuda5.5, so it is not working using...