Skip to content

Commit 40070ea

Browse files
committed
Adding missing license to functions
Signed-off-by: Kieran Fraser <[email protected]>
1 parent 0d15d2f commit 40070ea

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

art/estimators/object_detection/pytorch_detection_transformer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -912,14 +912,14 @@ def _apply_resizing(
912912
"scores": torch.from_numpy(target["scores"]).type(torch.float).to(self.device),
913913
}
914914
)
915-
916915
return x, targets
917916

918917

919918
class NestedTensor:
920919
"""
921920
From DETR source: https://github.com/facebookresearch/detr
922921
(detr/util/misc.py)
922+
License: Apache 2.0
923923
"""
924924

925925
def __init__(self, tensors, mask: Optional["torch.Tensor"]):
@@ -953,6 +953,7 @@ def nested_tensor_from_tensor_list(tensor_list: Union[List, "torch.Tensor"]):
953953
"""
954954
From DETR source: https://github.com/facebookresearch/detr
955955
(detr/util/misc.py)
956+
License: Apache 2.0
956957
"""
957958
import torch
958959

0 commit comments

Comments
 (0)