finished MDN-RNN and need to implement VAE

This commit is contained in:
junhong
2018-04-04 22:43:48 -04:00
parent 076e45cac0
commit 21d1d5dd8d
2 changed files with 72 additions and 11 deletions
+12
View File
@@ -0,0 +1,12 @@
import torch
from torch.nn import functional as F
import torch.nn as nn
class VAE(nn.Module):
def __init__(self):
# TODO: Variational Auto Encoder
super(VAE, self).__init__()
def forward(self, x):
pass