fix

7 months ago · 17bb7aa882
parent 7d077571cf
commit 17bb7aa882
1 changed files with 1 additions and 0 deletions
--- a/src/models/init.py
+++ b/src/models/init.py
@ -20,6 +20,7 @@ AVAILABLE_MODELS = {
    'SDYNetx1': sdynet.SDYNetx1,
    'RCNetx2Unlutable': rcnet.RCNetx2Unlutable,
    'RCNetx2CenteredUnlutable': rcnet.RCNetx2CenteredUnlutable,
+    'RCNetCentered_7x7v2': rcnet.RCNetCentered_7x7v2
 }

 def SaveCheckpoint(model, path):