Skip to content

Commit 316892c

Browse files
committed
enable deelab v3 TC
Signed-off-by: jiseong.oh <jiseong.oh@samsung.com>
1 parent 0f5befb commit 316892c

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

backends/samsung/test/models/test_deeplab_v3.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414

1515

1616
class TestMilestoneDeepLabV3(unittest.TestCase):
17-
@unittest.skip("Need to be fixed issue")
1817
def test_dl3_fp16(self):
1918
model = DeepLabV3ResNet50Model().get_eager_model()
2019
example_input = DeepLabV3ResNet50Model().get_example_inputs()

backends/samsung/test/utils/runtime_executor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,14 +142,14 @@ def run_on_device(self) -> Tuple[torch.Tensor]:
142142
else:
143143
break
144144

145-
num_of_output_files = len(os.listdir(host_output_save_dir))
145+
target_output_save_dir = os.path.join(host_output_save_dir, "output")
146+
num_of_output_files = len(os.listdir(target_output_save_dir))
146147
assert num_of_output_files == len(
147148
model_outputs
148149
), f"Number of outputs is invalid, expect {len(model_outputs)} while got {num_of_output_files}"
149150

150151
result = []
151152
for idx in range(num_of_output_files):
152-
target_output_save_dir = os.path.join(host_output_save_dir, "output")
153153
output_array = np.fromfile(
154154
os.path.join(target_output_save_dir, f"output_{idx}.bin"),
155155
dtype=np.uint8,

0 commit comments

Comments
 (0)