James Betker 8b376e63d9 More improvements		2022-03-16 10:16:34 -06:00
..
audio	More improvements	2022-03-16 10:16:34 -06:00
byol	Fix byol_model_wrapper to function with audio inputs	2021-08-05 22:20:22 -06:00
classifiers	Fix error & add nonfinite warning	2021-11-09 23:58:41 -07:00
clip	Refactor audio-style models into the audio folder	2022-03-15 11:06:25 -06:00
composable	composable initial checkin	2022-03-15 10:35:40 -06:00
diffusion	move diffusion vocoder	2022-03-08 15:51:05 -07:00
fixup_resnet	More refactoring	2020-12-18 09:18:34 -07:00
glean	Glean mods	2020-12-27 12:25:06 -07:00
lucidrains	new diffusion updates from testing	2022-01-29 11:01:01 -07:00
optical_flow	Add PWCNet for human optical flow	2021-01-25 08:25:44 -07:00
segformer	Fix error & add nonfinite warning	2021-11-09 23:58:41 -07: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
vqvae	asdf	2022-03-03 13:43:44 -07:00
__init__.py	Lots of new discriminator nets	2020-11-10 16:06:54 -07:00
arch_util.py	Update LR layers to checkpoint better	2022-01-22 08:22:57 -07: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	Fix codes when inferring from dvae	2021-10-17 22:51:17 -06:00
feature_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
lightweight_gan.py	Support functionality	2022-03-03 21:52:16 -07: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