We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 491bc03 commit 86576a1Copy full SHA for 86576a1
graph_net/paddle/validate.py
@@ -66,13 +66,13 @@ def main(args):
66
params.update(inputs)
67
state_dict = {k: utils.replay_tensor(v) for k, v in params.items()}
68
69
- y = model(**state_dict)[0]
+ y = model(**state_dict)
70
71
- print(np.argmin(y), np.argmax(y))
+ # print(np.argmin(y), np.argmax(y))
72
if isinstance(y, paddle.Tensor):
73
print(y.shape)
74
elif isinstance(y, list) or isinstance(y, tuple):
75
- print(y[0].shape if isinstance(y[0], paddle.tensor) else y[0])
+ print(y[0].shape if isinstance(y[0], paddle.Tensor) else y[0])
76
else:
77
raise ValueError("Illegal return value.")
78
0 commit comments