RuntimeError: reciprocal is not implemented for type torch.cuda.LongTensor


運行Faster rcnn FPN網絡,出現的bug:
RuntimeError: reciprocal is not implemented for type torch.cuda.LongTensor

 解決辦法: anchor_target_layer_fpn.py", line 136:

            num_examples = torch.sum(labels[i] >= 0)
            num_examples = num_examples.float() 
            positive_weights = 1.0 / num_examples
            negative_weights = 1.0 / num_examples

ref:https://blog.csdn.net/qq_43088746/article/details/100422779


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM