您好，我也是一样的情况，在出现了Namespace(backbone='vgg16_bn', gpu_id=0, line=2, output_dir='./logs/', row=2, weight_path='./weights/SHTechA.pth')之后，后面就没有反应了。不知道有没有顺利解决这个问题的

TencentYoutuResearch / CrowdCounting-P2PNet

The official codes for the ICCV2021 Oral presentation "Rethinking Counting and Localization in Crowds: A Purely Point-Based Framework"

Other

404 stars 109 forks source link

您好，我也是一样的情况，在出现了Namespace(backbone='vgg16_bn', gpu_id=0, line=2, output_dir='./logs/', row=2, weight_path='./weights/SHTechA.pth')之后，后面就没有反应了。不知道有没有顺利解决这个问题的 #49

Closed fatakWang closed 1 year ago

fatakWang commented 1 year ago

    您好，我也是一样的情况，在出现了Namespace(backbone='vgg16_bn', gpu_id=0, line=2, output_dir='./logs/', row=2, weight_path='./weights/SHTechA.pth')之后，后面就没有反应了。不知道有没有顺利解决这个问题的

Originally posted by @fatakWang in https://github.com/TencentYoutuResearch/CrowdCounting-P2PNet/issues/16#issuecomment-1364650983

fatakWang commented 1 year ago

我已经解决了这个问题，问题出在我的cuda版本安装的是11.6，然而我却按照readme的指示安装了pytorch1.5的版本，然而pytorch1.5只支持cuda10.2的，所以在model.to(cuda)时卡住了，在我使用与cuda11.6对应的pytorch后，并且在目录下修建一个logs的文件夹之后，就可以跑了

qiqihuang commented 1 year ago

楼上正解