Add SelecSLS to sotabench list

5 years ago · 84ca3d1f4d
parent 0062c15fb0
commit 84ca3d1f4d
1 changed files with 7 additions and 0 deletions
--- a/sotabench.py
+++ b/sotabench.py
@ -316,6 +316,13 @@ model_list = [
    _entry('hrnet_w44', 'HRNet-W44-C', '1908.07919'),
    _entry('hrnet_w48', 'HRNet-W48-C', '1908.07919'),
    _entry('hrnet_w64', 'HRNet-W64-C', '1908.07919'),
+
+
+    ## SelecSLS official impl weights
+    _entry('selecsls42b', 'SelecSLS-42_B', '1907.00837',
+           model_desc='Originally from https://github.com/mehtadushy/SelecSLS-Pytorch'),
+    _entry('selecsls60b', 'SelecSLS-60_B', '1907.00837',
+           model_desc='Originally from https://github.com/mehtadushy/SelecSLS-Pytorch'),
 ]

 for m in model_list: