GATConv.edge_updater = GATConv._orig_edge_updater model = torch.load(path) model = torch.load(path, weights_only=False) torch.jit.script(model) 2 changes: 1 addition & 1 deletion 2 test/nn/models/test_basic_gnn.py Original file line numberDiff line numberDiff line change @@ -2...