File name
Commit message
Commit date
import torch
import numpy as np
import pandas as pd
import plotly.express as px
from model import Autoencoder
from model import Generator
from model import Discriminator
from model import AttentiveRNN
## 대충 열심히 GAN 구성하는 코드
## 대충 그래서 weight export해서 inference용과 training용으로 나누는 코드
## 대충 그래서 inference용은 attention map까지 하는 녀석과 deraining까지 하는 녀석 두개가 나오는 코드
## 학습용은 그래서 풀 weight 나옴
## GAN은 학습 시키면 Nash equilibrium 나오는데 이거 ... 이번 프로그램에서는 문제가 안될려나?
##
## 대충 학습은 어떻게 돌려야 되지 하는 코드
## generator에서 튀어 나온 애들을 따로 저장해야 하는건가