mirror of
https://github.com/xuebinqin/DIS.git
synced 2024-11-26 08:43:17 +01:00
Update isnet.py
This commit is contained in:
parent
10657f42e5
commit
996c9495d4
@ -544,7 +544,7 @@ class ISNetDIS(nn.Module):
|
||||
hx = x
|
||||
|
||||
hxin = self.conv_in(hx)
|
||||
hx = self.pool_in(hxin)
|
||||
#hx = self.pool_in(hxin)
|
||||
|
||||
#stage 1
|
||||
hx1 = self.stage1(hxin)
|
||||
|
Loading…
Reference in New Issue
Block a user