I want to test the GIN model. Therefore, I downloaded "gin_best.pth" from "DIG_storage/xgraph/checkpoint/mutag/", and I used torch.load() to load it as following codes.
from dig.xgraph.models import GIN_3l model = GIN_3l(model_level='node', dim_node=dim_node, dim_hidden=128, num_classes=num_classes) model.load_state_dict(torch.load(pth_path, map_location=torch.device('cpu')))
It seems not match, and where can I get the model matched this "pth" file.
I want to test the GIN model. Therefore, I downloaded "gin_best.pth" from "DIG_storage/xgraph/checkpoint/mutag/", and I used torch.load() to load it as following codes.
from dig.xgraph.models import GIN_3l model = GIN_3l(model_level='node', dim_node=dim_node, dim_hidden=128, num_classes=num_classes) model.load_state_dict(torch.load(pth_path, map_location=torch.device('cpu')))It seems not match, and where can I get the model matched this "pth" file.