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에서 튀어 나온 애들을 따로 저장해야 하는건가