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