Skip to content

Commit 6925062

Browse files
leeclemnetclaude
andcommitted
Point NAS path templates at the verified bucket layout
Lee copied the artifacts to <family>/<config>/inference_model.onnx with no nas/ prefix; verified by listing the bucket (6188 configs per family). Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
1 parent cd34f39 commit 6925062

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

scripts/benchmark_platform_nas_models.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def to_requests(self, buffer_time: float, max_images: int | None) -> list[Artifa
111111
NasFamily(
112112
family="rfdetr-nas-parent",
113113
inference_class=RFDETRTRTInference,
114-
onnx_path_template="nas/rfdetr-nas-parent/{config_id}/inference_model.onnx",
114+
onnx_path_template="rfdetr-nas-parent/{config_id}/inference_model.onnx",
115115
configs=(
116116
(384, 16, 2, 2, 300), # rfdetr-nano
117117
(512, 16, 2, 3, 300), # rfdetr-small
@@ -122,7 +122,7 @@ def to_requests(self, buffer_time: float, max_images: int | None) -> list[Artifa
122122
NasFamily(
123123
family="rfdetr-nas-base-parent",
124124
inference_class=RFDETRTRTInference,
125-
onnx_path_template="nas/rfdetr-nas-base-parent/{config_id}/inference_model.onnx",
125+
onnx_path_template="rfdetr-nas-base-parent/{config_id}/inference_model.onnx",
126126
configs=(
127127
(700, 20, 1, 5, 300), # rfdetr-xlarge
128128
(880, 20, 2, 5, 300), # rfdetr-2xlarge
@@ -131,15 +131,15 @@ def to_requests(self, buffer_time: float, max_images: int | None) -> list[Artifa
131131
NasFamily(
132132
family="rfdetr-nas-pecoret-parent",
133133
inference_class=RFDETRTRTInference,
134-
onnx_path_template="nas/rfdetr-nas-pecoret-parent/{config_id}/inference_model.onnx",
134+
onnx_path_template="rfdetr-nas-pecoret-parent/{config_id}/inference_model.onnx",
135135
# No named model uses the PE-Core-T backbone, so this family has no
136136
# named-equivalent tuple to start from. Add the tuples this sweep needs.
137137
configs=(),
138138
),
139139
NasFamily(
140140
family="rfdetr-nas-seg-parent",
141141
inference_class=RFDETRSegTRTInference,
142-
onnx_path_template="nas/rfdetr-nas-seg-parent/{config_id}/inference_model.onnx",
142+
onnx_path_template="rfdetr-nas-seg-parent/{config_id}/inference_model.onnx",
143143
configs=(
144144
(312, 12, 1, 4, 100), # rfdetr-seg-nano
145145
(384, 12, 2, 4, 100), # rfdetr-seg-small

tests/test_platform_model_inventory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def test_seg_family_runs_the_seg_inference_class():
134134

135135
def test_onnx_path_follows_the_family_template():
136136
family = next(f for f in nas.NAS_FAMILIES if f.family == "rfdetr-nas-parent")
137-
assert family.onnx_path((512, 16, 2, 3, 300)) == "nas/rfdetr-nas-parent/512_16_2_3_300/inference_model.onnx"
137+
assert family.onnx_path((512, 16, 2, 3, 300)) == "rfdetr-nas-parent/512_16_2_3_300/inference_model.onnx"
138138

139139

140140
def test_a_wrong_path_template_stops_the_run_before_it_starts(monkeypatch):
@@ -144,7 +144,7 @@ def test_a_wrong_path_template_stops_the_run_before_it_starts(monkeypatch):
144144
monkeypatch.setattr(nas.os.path, "exists", lambda path: False)
145145
missing = nas.missing_artifacts(all_nas_requests())
146146
assert len(missing) == len(all_nas_requests())
147-
assert "nas/rfdetr-nas-parent/384_16_2_2_300/inference_model.onnx" in missing
147+
assert "rfdetr-nas-parent/384_16_2_2_300/inference_model.onnx" in missing
148148

149149

150150
def test_nas_rows_spell_the_tuple_out():

0 commit comments

Comments
 (0)