Commit c2b1680f authored by JingweiZhang12's avatar JingweiZhang12 Committed by ZwwWayne
Browse files

add args and kwargs

parent adb17824
...@@ -360,7 +360,7 @@ class BaseInstance3DBoxes(object): ...@@ -360,7 +360,7 @@ class BaseInstance3DBoxes(object):
with_yaw=boxes_list[0].with_yaw) with_yaw=boxes_list[0].with_yaw)
return cat_boxes return cat_boxes
def to(self, device): def to(self, device, *args, **kwargs):
"""Convert current boxes to a specific device. """Convert current boxes to a specific device.
Args: Args:
...@@ -372,7 +372,7 @@ class BaseInstance3DBoxes(object): ...@@ -372,7 +372,7 @@ class BaseInstance3DBoxes(object):
""" """
original_type = type(self) original_type = type(self)
return original_type( return original_type(
self.tensor.to(device), self.tensor.to(device, *args, **kwargs),
box_dim=self.box_dim, box_dim=self.box_dim,
with_yaw=self.with_yaw) with_yaw=self.with_yaw)
......
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