|
|
@ -220,9 +220,7 @@ def infer(use_cuda, save_dirname=None):
|
|
|
|
np_data = np.array(results[0])
|
|
|
|
np_data = np.array(results[0])
|
|
|
|
infer_out = infer_outputs[0].data.float_data()
|
|
|
|
infer_out = infer_outputs[0].data.float_data()
|
|
|
|
for a, b in zip(np_data[0], infer_out):
|
|
|
|
for a, b in zip(np_data[0], infer_out):
|
|
|
|
g_a = float("{:.6g}".format(a))
|
|
|
|
assert np.isclose(a, b), "a: {}, b: {}".format(a, b)
|
|
|
|
g_b = float("{:.6g}".format(b))
|
|
|
|
|
|
|
|
assert g_a == g_b
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main(use_cuda, is_sparse, is_parallel):
|
|
|
|
def main(use_cuda, is_sparse, is_parallel):
|
|
|
|