import megengine as mge
import megengine.module as M
import megengine.functional as F
from megengine.core import Parameter
from utils import *

def addLeakyRelu(x):
    return M.Sequential(x, M.LeakyReLU(0.1))

def addSig(x):
    return M.Sequential(x, M.Sigmoid())

def up_block(x, ic, oc):
    return 
M.ConvTranspose2d(ic, oc, 4, stride=2, padding=1)