Update model/modnet.py

model/modnet.py

@@ -7,7 +7,7 @@ import torch.nn as nn
 
 
 class MODNet(nn.Module):
-    def __init__(self):
+    def __init__(self, in_channels=3, hr_channels=32, backbone_arch='mobilenetv2', backbone_pretrained=True):
         super(MODNet, self).__init__()
         self.backbone = nn.Identity()  # Replaced with identity for fast test