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):