diff --git a/test/external/mlperf_retinanet/coco_utils.py b/test/external/mlperf_retinanet/coco_utils.py index 5ab1b3b18d..756bd6955d 100644 --- a/test/external/mlperf_retinanet/coco_utils.py +++ b/test/external/mlperf_retinanet/coco_utils.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/coco_utils.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/coco_utils.py import os import torch diff --git a/test/external/mlperf_retinanet/model/boxes.py b/test/external/mlperf_retinanet/model/boxes.py index f98effd1ef..96dbceeb95 100644 --- a/test/external/mlperf_retinanet/model/boxes.py +++ b/test/external/mlperf_retinanet/model/boxes.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/boxes.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/boxes.py import torch from torch import Tensor diff --git a/test/external/mlperf_retinanet/model/image_list.py b/test/external/mlperf_retinanet/model/image_list.py index 426d575b6a..f7a4cf2eb9 100644 --- a/test/external/mlperf_retinanet/model/image_list.py +++ b/test/external/mlperf_retinanet/model/image_list.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/image_list.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/image_list.py import torch from torch import Tensor diff --git a/test/external/mlperf_retinanet/model/transform.py b/test/external/mlperf_retinanet/model/transform.py index dad96bf320..b68e8cfb2d 100644 --- a/test/external/mlperf_retinanet/model/transform.py +++ b/test/external/mlperf_retinanet/model/transform.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/transform.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/transform.py import torch diff --git a/test/external/mlperf_retinanet/model/utils.py b/test/external/mlperf_retinanet/model/utils.py index d976e1955a..33c64c16cb 100644 --- a/test/external/mlperf_retinanet/model/utils.py +++ b/test/external/mlperf_retinanet/model/utils.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/utils.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/model/utils.py import torch diff --git a/test/external/mlperf_retinanet/presets.py b/test/external/mlperf_retinanet/presets.py index 5d45313c5f..fb918581b4 100644 --- a/test/external/mlperf_retinanet/presets.py +++ b/test/external/mlperf_retinanet/presets.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/presets.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/presets.py from test.external.mlperf_retinanet import transforms as T diff --git a/test/external/mlperf_retinanet/transforms.py b/test/external/mlperf_retinanet/transforms.py index c37bf84767..0a35347d1c 100644 --- a/test/external/mlperf_retinanet/transforms.py +++ b/test/external/mlperf_retinanet/transforms.py @@ -1,4 +1,4 @@ -# https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/transforms.py +# Copied from https://github.com/mlcommons/training/blob/637c82f9e699cd6caf108f92efb2c1d446b630e0/single_stage_detector/ssd/transforms.py import torch import torchvision