Setup Relaxed Group Convolutional Network - Githubissues

dgcnz / dl2

Code for "Effect of equivariance on training dynamics"

1 stars 0 forks source link

Setup Relaxed Group Convolutional Network #34

Closed dgcnz closed 1 month ago

dgcnz commented 1 month ago

What does this PR do?

Features:

[x] Adds RelaxedGCNN

Tests:

[x] Forward correctness
[x] Correctness on changes
[x] Benchmark on changes

Fixes #4

Before submitting

[x] Did you make sure title is self-explanatory and the description concisely explains the PR?
[x] Did you make sure your PR does only one thing, instead of bundling different changes together?
[x] Did you list all the breaking changes introduced by this pull request?
[x] Did you test your PR locally with pytest command?
[ ] Did you run pre-commit hooks with pre-commit run -a command?