FUNDED_NISL
FUNDED_NISL copied to clipboard
Loading pre-trained model weights fails.
Hello! When I 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:
It mentioned
error! in traububg_utils.py line199
. Does it mean cli_data_hyperparameter_overrides=args.data_param_override, line 199
in file training_utils.py
? So how can I solve this problem? And is this the reason why loading pre-trained model weights fails? Looking forward to your reply : ).