File main.py

tutorials/03-advanced/variational_autoencoder/main.py:None–None

import os
import torch
import torch.nn as nn
import torch.nn.functional as F