未来不再遥远

import torch import torch.nn as nn class 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 nn class 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...