From e6389184c55743892b020b9d66e47e15794a0355 Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Fri, 21 Mar 2025 15:07:45 -0400 Subject: [PATCH] update comment for retinanet dataloader implementations (#9534) Co-authored-by: chenyu --- test/external/mlperf_retinanet/coco_utils.py | 2 +- test/external/mlperf_retinanet/model/boxes.py | 2 +- test/external/mlperf_retinanet/model/image_list.py | 2 +- test/external/mlperf_retinanet/model/transform.py | 2 +- test/external/mlperf_retinanet/model/utils.py | 2 +- test/external/mlperf_retinanet/presets.py | 2 +- test/external/mlperf_retinanet/transforms.py | 2 +- 7 files changed, 7 insertions(+), 7 deletions(-) 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