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.
2 parents 4185be4 + b3fb4bf commit cd6a548Copy full SHA for cd6a548
graph_net/paddle/validate.py
@@ -12,6 +12,7 @@
12
import graph_net
13
import os
14
import re
15
+import paddle
16
17
18
def load_class_from_file(file_path: str, class_name: str):
@@ -68,7 +69,12 @@ def main(args):
68
69
y = model(**state_dict)[0]
70
71
print(np.argmin(y), np.argmax(y))
- print(y.shape)
72
+ if isinstance(y, paddle.Tensor):
73
+ print(y.shape)
74
+ elif isinstance(y, list) or isinstance(y, tuple):
75
+ print(y[0].shape)
76
+ else:
77
+ raise ValueError("Illegal Return Value.")
78
79
if not args.no_check_redundancy:
80
print("Check redundancy ...")
0 commit comments