修正的轻量型UNet网络

import torch.nn as nn
import torch

from tensorboardX import SummaryWriter

'''
2021/6/1
problem
1.nn.Batch
2.Conv2dtranpose-nn.ConvTranspose2d(in_channels=out_c,out_channels=out_c,kernel_size=4,stride=2)
3.dropout(p)
4.deblock (2*in_c,in_c,out_c)??
'''

class block(nn