diff --git a/semtorch/models/archs/backbones/build.py b/semtorch/models/archs/backbones/build.py
index 1a1d4eb..35fae93 100644
--- a/semtorch/models/archs/backbones/build.py
+++ b/semtorch/models/archs/backbones/build.py
@@ -22,8 +22,8 @@
     'resnet152c': 'https://github.com/LikeLy-Journey/SegmenTron/releases/download/v0.1.0/resnet152-0d43d698.pth',
     'xception65': 'https://github.com/LikeLy-Journey/SegmenTron/releases/download/v0.1.0/tf-xception65-270e81cf.pth',
     'mobilenet_v2': 'https://github.com/LikeLy-Journey/SegmenTron/releases/download/v0.1.0/mobilenetV2-15498621.pth',
-    "hrnet_w18_small_v1": "https://dl.dropboxusercontent.com/s/do1we7sfxowkm2w/hrnet_w18_small_model_v1.pth",
-    "hrnet_w18_small_v2": "https://dl.dropboxusercontent.com/s/pz3vdp8jg0ffkbr/hrnet_w18_small_model_v2.pth",
+    "hrnet_w18_small_model_v1": "https://dl.dropboxusercontent.com/s/do1we7sfxowkm2w/hrnet_w18_small_model_v1.pth",
+    "hrnet_w18_small_model_v2": "https://dl.dropboxusercontent.com/s/pz3vdp8jg0ffkbr/hrnet_w18_small_model_v2.pth",
     "hrnet_w18": "https://dl.dropboxusercontent.com/s/cewi4owfrw00oza/hrnetv2_w18_imagenet_pretrained.pth",
     "hrnet_w30": "https://dl.dropboxusercontent.com/s/r1rafhhw1hfhpgl/hrnetv2_w30_imagenet_pretrained.pth",
     "hrnet_w32": "https://dl.dropboxusercontent.com/s/7bu7mku682von0f/hrnetv2_w32_imagenet_pretrained.pth",
diff --git a/semtorch/models/archs/backbones/hrnet.py b/semtorch/models/archs/backbones/hrnet.py
index 6c47603..781c42c 100644
--- a/semtorch/models/archs/backbones/hrnet.py
+++ b/semtorch/models/archs/backbones/hrnet.py
@@ -476,7 +476,7 @@ def init_weights(self, pretrained=''):
         return "HRNet backbone wieghts loaded"
 
 backbone_config={
-    "hrnet_w18_small_v1": {
+    "hrnet_w18_small_model_v1": {
         "STAGE1": {
             "NUM_MODULES": 1,
             "NUM_BRANCHES": 1,
@@ -510,7 +510,7 @@ def init_weights(self, pretrained=''):
             "FUSE_METHOD": "SUM"
         }
     },
-    "hrnet_w18_small_v2": {
+    "hrnet_w18_small_model_v2": {
         "STAGE1": {
             "NUM_MODULES": 1,
             "NUM_BRANCHES": 1,
@@ -683,13 +683,13 @@ def init_weights(self, pretrained=''):
 }
 
 @BACKBONE_REGISTRY.register()
-def hrnet_w18_small_v1(norm_layer=nn.BatchNorm2d):
-    model = HighResolutionNet(config=backbone_config["hrnet_w18_small_v1"], norm_layer=norm_layer)
+def hrnet_w18_small_model_v1(norm_layer=nn.BatchNorm2d):
+    model = HighResolutionNet(config=backbone_config["hrnet_w18_small_model_v1"], norm_layer=norm_layer)
     return model
 
 @BACKBONE_REGISTRY.register()
-def hrnet_w18_small_v2(norm_layer=nn.BatchNorm2d):
-    model = HighResolutionNet(config=backbone_config["hrnet_w18_small_v2"], norm_layer=norm_layer)
+def hrnet_w18_small_model_v2(norm_layer=nn.BatchNorm2d):
+    model = HighResolutionNet(config=backbone_config["hrnet_w18_small_model_v2"], norm_layer=norm_layer)
     return model
 
 @BACKBONE_REGISTRY.register()
diff --git a/semtorch/models/archs/hrnet_seg.py b/semtorch/models/archs/hrnet_seg.py
index 97c20be..97b500a 100644
--- a/semtorch/models/archs/hrnet_seg.py
+++ b/semtorch/models/archs/hrnet_seg.py
@@ -18,10 +18,10 @@
 
 
 architecture_config = {
-    "hrnet_w18_small_v1": {
+    "hrnet_w18_small_model_v1": {
         "FINAL_CONV_KERNEL": 1
     },
-    "hrnet_w18_small_v2": {
+    "hrnet_w18_small_model_v2": {
         "FINAL_CONV_KERNEL": 1
     },
     "hrnet_w18": {