Update model/HWMNet.py
Browse files- model/HWMNet.py +0 -1
model/HWMNet.py
CHANGED
|
@@ -267,7 +267,6 @@ class HWMNet(nn.Module):
|
|
| 267 |
return out_1
|
| 268 |
|
| 269 |
if __name__ == "__main__":
|
| 270 |
-
from thop import profile
|
| 271 |
input = torch.ones(1, 3, 400, 592, dtype=torch.float, requires_grad=False).cuda()
|
| 272 |
|
| 273 |
model = HWMNet(in_chn=3, wf=96, depth=4).cuda()
|
|
|
|
| 267 |
return out_1
|
| 268 |
|
| 269 |
if __name__ == "__main__":
|
|
|
|
| 270 |
input = torch.ones(1, 3, 400, 592, dtype=torch.float, requires_grad=False).cuda()
|
| 271 |
|
| 272 |
model = HWMNet(in_chn=3, wf=96, depth=4).cuda()
|