Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

if you encounter loading pre-trained model weights fails.You can borrow my method. #30

Open
TintinDevCoder opened this issue Oct 26, 2024 · 1 comment

Comments

@TintinDevCoder
Copy link

Hi, I found a little problem.
When try to follow the command CUDA_VISIBLE_DEVICES=2 python test.py GGNN GraphBinaryClassification ../data/data/data/cve/badall --storedModel_path "./trained_model/GGNN_GraphBinaryClassification__2023-02-01_05-36-00_f1 = 0.800_best.pkl", one issue occurred. The details are as follows:
329828893-a4c0f830-ea93-422b-8c5f-c45506187746
You just need to unzip the pre-processed CWE-77.zip file located in the "FUNDED/data/data" directory, and rename the 5result file to CWE-77.like this:
1729905268911

@HuantWang
Copy link
Owner

What if you don't change any folder names? If everything works fine until the folder name is changed, it might be due to a file path issue.

Alternatively, check the package versions listed in requirements.txt to ensure they match. Since this project was constructed a long time ago, some packages may have been updated, which could cause compatibility issues.

Please let me know what happens next.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants