James Betker 33120cb35c Add norming to discretization_loss		2021-10-06 17:10:50 -06:00
..
__init__.py	Add dvae with channel attention	2021-10-03 10:52:01 -06:00
attention_dvae.py	Add dvae with channel attention	2021-10-03 10:52:01 -06:00
discretization_loss.py	Add norming to discretization_loss	2021-10-06 17:10:50 -06:00
dvae_channel_attention.py	Allow attention_dim in channel attention to be specified, add converter	2021-10-05 17:29:38 -06:00