You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The current implementation, such as cuda only supports Tensor, BaseDataElemenet, which doesn't support data strucutures such as BaseInstance3DBoxes. See the below code:
defcuda(self) ->'BaseDataElement':
"""Convert all tensors to GPU in data."""new_data=self.new()
fork, vinself.items():
ifisinstance(v, (torch.Tensor, BaseDataElement)):
v=v.cuda()
data= {k: v}
new_data.set_data(data)
returnnew_data
why not changing the implementation to work with any data strucutures that has a method of cuda?
Any other context?
No response
The text was updated successfully, but these errors were encountered:
What is the feature?
The current implementation, such as
cuda
only supportsTensor, BaseDataElemenet
, which doesn't support data strucutures such asBaseInstance3DBoxes
. See the below code:why not changing the implementation to work with any data strucutures that has a method of
cuda
?Any other context?
No response
The text was updated successfully, but these errors were encountered: