import torch.nn as nn. import torch. class UNet(nn.Module): ... Conv2d(128, 128, kernel_size = 3, padding = 1). self.pool2 = nn.
確定! 回上一頁