zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Sync

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-03 23:13:57 +00:00
parent da8287b7e8
commit c74f4c7fb3
3 changed files with 17 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

1
model/stn.py
View file

@ -24,7 +24,6 @@ class STNet(nn.Module):
_dummy_size_ = input_size
# shape checking
print("STN: dummy_size {}".format(_dummy_size_))
_dummy_x_ = torch.zeros(_dummy_size_)
# (3.1) Spatial transformer localization-network