James Betker
|
4d1a42e944
|
Add switchnorm to gumbel_quantizer
|
2021-09-24 18:49:25 -06:00 |
|
James Betker
|
061dbcd458
|
Another fix to anorm
|
2021-06-06 15:09:49 -06:00 |
|
James Betker
|
9a6991e461
|
Fix switch norm average
|
2021-06-06 15:04:28 -06:00 |
|
James Betker
|
57e1a6a0f2
|
cifar: add hard routing
Also mods switched_routing to support non-pixular inputs
|
2021-06-06 14:53:43 -06:00 |
|
James Betker
|
9fc3df3f5b
|
Switched conv: add conversion function with allowlist
|
2021-03-13 10:44:56 -07:00 |
|
James Betker
|
cf9a6da889
|
Fix some bugs, checkin work on vqvae3
|
2021-03-02 20:56:19 -07:00 |
|
James Betker
|
e7be4bdff3
|
Revert
|
2021-02-05 08:43:07 -07:00 |
|
James Betker
|
6dec1f5968
|
Back to groupnorm
|
2021-02-05 08:42:11 -07:00 |
|
James Betker
|
336f807c8e
|
lambda2
|
2021-02-05 00:00:24 -07:00 |
|
James Betker
|
025a5867c4
|
Use syncbatchnorm instead
|
2021-02-04 22:26:36 -07:00 |
|
James Betker
|
bb79fafb89
|
Fix groupnorm specification
|
2021-02-04 22:15:38 -07:00 |
|
James Betker
|
43da1f9c4b
|
Convert lambda coupler to use groupnorm instead of batchnorm
|
2021-02-04 21:59:44 -07:00 |
|
James Betker
|
7070142805
|
Make vqvae3_hard more configurable
|
2021-02-04 09:03:22 -07:00 |
|
James Betker
|
1405ff06b8
|
Fix SwitchedConvHardRoutingFunction for current cuda router
|
2021-02-03 14:11:55 -07:00 |
|
James Betker
|
320edbaa3c
|
Move switched_conv logic around a bit
|
2021-02-02 20:41:24 -07:00 |
|
James Betker
|
07168ecfb4
|
Enable vqvae to use a switched_conv variant
|
2021-01-09 20:53:14 -07:00 |
|
James Betker
|
5640e4efe4
|
More refactoring
|
2020-12-18 09:18:34 -07:00 |
|