James Betker f2977d360c Allow attention_dim in channel attention to be specified, add converter		2021-10-05 17:29:38 -06:00
..
byol	Fix byol_model_wrapper to function with audio inputs	2021-08-05 22:20:22 -06:00
classifiers	Various mods to support better jpeg image filtering	2021-06-25 13:16:15 -06:00
diffusion	Increase baseline codes recording across all dvae models	2021-09-30 08:09:07 -06:00
fixup_resnet	More refactoring	2020-12-18 09:18:34 -07:00
flownet2@db2b7899ea
glean	Glean mods	2020-12-27 12:25:06 -07:00
gpt_voice	Allow attention_dim in channel attention to be specified, add converter	2021-10-05 17:29:38 -06:00
optical_flow	Add PWCNet for human optical flow	2021-01-25 08:25:44 -07:00
segformer	Also debug distribution of switch	2021-06-07 15:36:07 -06:00
spleeter	Improvements to splitter	2021-09-09 23:34:56 -06:00
srflow	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
stylegan	Various fixes	2021-07-14 00:08:42 -06:00
switched_conv	Add switchnorm to gumbel_quantizer	2021-09-24 18:49:25 -06:00
tacotron2	Stop dataset - attempt #2	2021-08-18 18:29:38 -06:00
vqvae	Discretization loss attempt	2021-10-04 20:59:21 -06:00
waveglow	Add waveglow & inference capabilities to audio generator	2021-07-08 23:07:36 -06:00
__init__.py
arch_util.py	More work on wave-diffusion	2021-07-27 05:36:17 -06:00
audio_resnet.py	Add audio augmentation to wavfile_dataset, utility to test audio similary	2021-08-05 22:14:49 -06:00
discriminator_vgg_arch.py	Misc fixes	2021-01-04 10:53:53 -07:00
feature_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
lightweight_gan.py	Various fixes	2021-07-14 00:08:42 -06:00
ResGen_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
RRDBNet_arch.py	More cleanup	2021-09-29 14:24:49 -06:00
spinenet_arch.py	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00