Commit 986d1f67 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

update np_example_to_features

parent 9a97d0d7
......@@ -120,7 +120,7 @@ def np_example_to_features(
dtype=torch.float32,
)
return {k: v for k, v in features.items()},gt_features
return {k: v for k, v in features.items()}, gt_features
class FeaturePipeline:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment