未来不再遥远

import torchimport torch.nn as nnclass pub(nn.Module): def __init__(self, in_channel, out_channel, batch_norm=True, keep_size=False): super(pub, self).__init__() pad = 1 if kee...

from torch import nnclass pub(nn.Module): def __init__(self, in_channels, out_channels, batch_norm=True): super(pub, self).__init__() inter_channels = in_channels/2 if in_chann...