File size: 261 Bytes
f87c369
 
 
 
 
 
5f566b3
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
# Transformer-VAE (WIP)

A PyTorch Transformer-VAE model.

Uses an MMD loss to prevent posterior collapse.

Will setup in the next month or so.

## ToDo
- [ ] Copy in old repo code.
- [ ] Make a bunch of sample training runs.
- [ ] Make an interpolation widget?