mirror of
https://github.com/wassname/world-models-pytorch.git
synced 2026-06-27 18:45:50 +08:00
junhong
13 lines
237 B
Python
13 lines
237 B
Python
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
|