Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement grouped convolutions #116

Merged
Merged
Changes from 1 commit
Commits
Show all changes
47 commits
Select commit Hold shift + click to select a range
01a425e
Add groups_of template parameter
purefunctor Nov 14, 2023
73edf91
Add test data for grouped convolutions
purefunctor Nov 14, 2023
2d8d034
Fix data generation script
purefunctor Nov 14, 2023
32dd08d
Initial working version for grouped convolutions
purefunctor Nov 15, 2023
8ba4df0
Specialize forward method when groups_of is 1
purefunctor Nov 15, 2023
a1b1958
Remove extra stuff in CMakeLists.txt
purefunctor Nov 15, 2023
ffc2d4f
Clean up conv1d.h
purefunctor Nov 15, 2023
1fa8bf7
Restore conditional for dynamic state
purefunctor Nov 15, 2023
cc5995e
Test different configuration
purefunctor Nov 20, 2023
0811e5b
Merge column copying logic
purefunctor Nov 20, 2023
8ba8901
Implement microtcn layer in PyTorch
purefunctor Nov 20, 2023
9d6545c
Try to implement microtcn
purefunctor Nov 20, 2023
e9e41ef
Make first half of microtcn work
purefunctor Nov 20, 2023
aa6dd34
Add more group tests
purefunctor Nov 20, 2023
5a3158b
Add .venv to gitignore
purefunctor Nov 20, 2023
ec2a0b4
New definition for causal_crop
purefunctor Nov 20, 2023
13411c9
TCNBlock works!!!
purefunctor Nov 20, 2023
3e0aba5
Merge branch 'main' into pure/implement-groups
purefunctor Nov 20, 2023
b4a0e60
Fix loadLayer to use groups_of
purefunctor Nov 20, 2023
37addf0
Bring back old tests
purefunctor Nov 20, 2023
8096969
Add groups_of template parameter
purefunctor Nov 14, 2023
e65a483
Add test data for grouped convolutions
purefunctor Nov 14, 2023
f4fa2f0
Fix data generation script
purefunctor Nov 14, 2023
88ccc7c
Initial working version for grouped convolutions
purefunctor Nov 15, 2023
b9c8451
Specialize forward method when groups_of is 1
purefunctor Nov 15, 2023
b3b0345
Remove extra stuff in CMakeLists.txt
purefunctor Nov 15, 2023
4ba2217
Clean up conv1d.h
purefunctor Nov 15, 2023
6b5d11a
Restore conditional for dynamic state
purefunctor Nov 15, 2023
dbf2365
Test different configuration
purefunctor Nov 20, 2023
e80579b
Merge column copying logic
purefunctor Nov 20, 2023
0d08dac
Implement microtcn layer in PyTorch
purefunctor Nov 20, 2023
8def3b6
Try to implement microtcn
purefunctor Nov 20, 2023
b3d430d
Make first half of microtcn work
purefunctor Nov 20, 2023
d23c357
Add more group tests
purefunctor Nov 20, 2023
3813f27
Add .venv to gitignore
purefunctor Nov 20, 2023
83b8aaf
New definition for causal_crop
purefunctor Nov 20, 2023
c20b09f
TCNBlock works!!!
purefunctor Nov 20, 2023
c3c0880
Fix loadLayer to use groups_of
purefunctor Nov 20, 2023
19208e9
Bring back old tests
purefunctor Nov 20, 2023
b9a2411
Tweaks for testing code
jatinchowdhury18 Nov 17, 2023
fede4e2
Fixes, Conv1D with groups works for everything except Eigen backend
jatinchowdhury18 Nov 27, 2023
4aa1d63
Adding Eigen implementation and fixing channel indexing
jatinchowdhury18 Nov 27, 2023
9fc5472
Re-add groups test
jatinchowdhury18 Nov 28, 2023
98d055e
Bring back MicroTCN test
jatinchowdhury18 Nov 28, 2023
44b1c4f
Merge remote-tracking branch 'upstream/purefunctor-pure/implement-gro…
purefunctor Nov 29, 2023
a48119c
Rename groups_of -> groups
purefunctor Nov 29, 2023
7aebcac
Also rename groups_of -> groups
purefunctor Nov 29, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
TCNBlock works!!!
purefunctor authored and jatinchowdhury18 committed Nov 28, 2023
commit c20b09fb8216a71a6bfc9ead6ec94ea2ae0b435f
10 changes: 5 additions & 5 deletions python/microtcn.py
Original file line number Diff line number Diff line change
@@ -57,14 +57,14 @@ def forward(self, x):
x_in = x

x = self.conv1(x)
# x = self.bn(x)
# x = self.relu(x)
x = self.bn(x)
x = self.relu(x)

# x_res = self.res(x_in)
x_res = self.res(x_in)

# x = x + causal_crop(x_res, x.shape[-1])
x = x + causal_crop(x_res, x.shape[-1])

return causal_crop(self.res(x_in), x.shape[-1])
return x

def export(self):
bn_dict = self.bn.state_dict()
Loading