lindsey98 / PhishIntention

PhishIntention: Phishing detection through webpage intention
MIT License
45 stars 12 forks source link

Code seems not working on CPU #9

Open imethanlee opened 2 years ago

imethanlee commented 2 years ago

It is not a big problem since I believe no one will run this code without GPU. However, it seems true that even if I install the CPU version of pytorch and detectron, the code still fails with the following output. Just raise this issue if any other come across the same problem.

image

lindsey98 commented 2 years ago

Hi, did you uncomment this line https://github.com/lindsey98/PhishIntention/blob/main/phishintention/src/AWL_detector.py#L32 ?

imethanlee commented 2 years ago

Uhhh......seems like I didn't notice that. It works now. Should also uncomment here https://github.com/lindsey98/PhishIntention/blob/main/phishintention/src/crp_locator.py#L43 to make it work on CPU. Thanks.