oldwolf204
Results
2
comments of
oldwolf204
我也遇到了同样的问题，楼上的有解决方案了吗？
modify modules.py code: class UNet(nn.Module): #modified by hx add param imgsize def __init__(self, c_in=3, c_out=3, imgsize=64,time_dim=256, device="cuda"): super().__init__() self.device = device self.time_dim = time_dim self.inc = DoubleConv(c_in, 64) self.down1 =...