Spaces:

52Hz
/

HWMNet_lowlight_enhancement

Runtime error

52Hz commited on Jan 27, 2022

Commit

dc67593

1 Parent(s): da67f47

Update model/HWMNet.py

Files changed (1) hide show

model/HWMNet.py CHANGED Viewed

@@ -267,7 +267,6 @@ class HWMNet(nn.Module):
         return out_1
 if __name__ == "__main__":
-    from thop import profile
     input = torch.ones(1, 3, 400, 592, dtype=torch.float, requires_grad=False).cuda()
     model = HWMNet(in_chn=3, wf=96, depth=4).cuda()

         return out_1
 if __name__ == "__main__":
     input = torch.ones(1, 3, 400, 592, dtype=torch.float, requires_grad=False).cuda()
     model = HWMNet(in_chn=3, wf=96, depth=4).cuda()