From 17bb7aa882317aabfdc1c22b0bea52a417b0f3ad Mon Sep 17 00:00:00 2001
From: Vladimir Protsenko <protsenkovi@gmail.com>
Date: Mon, 22 Apr 2024 14:14:15 +0000
Subject: [PATCH] fix

---
 src/models/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/models/__init__.py b/src/models/__init__.py
index 900d147..33b588d 100644
--- 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):