From 4573ec9247497564383db79363b4ab0b88de0de7 Mon Sep 17 00:00:00 2001 From: "Cano-Muniz, Santiago" Date: Sat, 8 Mar 2025 20:57:36 +0100 Subject: [PATCH] Clarify comments for RESTORMER_CONFIGS in test_restormer.py, I, Cano-Muniz, Santiago , hereby add my Signed-off-by to this commit: 3c2dbc61f26c16d411558f9adb0fd92c5a65569b Signed-off-by: Cano-Muniz, Santiago --- tests/networks/nets/test_restormer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/networks/nets/test_restormer.py b/tests/networks/nets/test_restormer.py index 9b54b7a765..f0466caa2d 100644 --- a/tests/networks/nets/test_restormer.py +++ b/tests/networks/nets/test_restormer.py @@ -40,10 +40,10 @@ ] RESTORMER_CONFIGS = [ - # 2-level architecture + # 2-level architecture test {"num_blocks": [1, 1], "heads": [1, 1]}, {"num_blocks": [2, 1], "heads": [2, 1]}, - # 3-level architecture + # 3-level architecture test {"num_blocks": [1, 1, 1], "heads": [1, 1, 1]}, {"num_blocks": [2, 1, 1], "heads": [2, 1, 1]}, ]