getting this error RuntimeError: mat1 and mat2 shapes cannot be multiplied (128x128 and 256x128)
when I am using the code below which is basically an array of size 32x32x32 and timesteps of 128:
size = 32
device_val="cuda"
x_input = torch.tensor(np.random.rand(1,size, size, size)).to(device_val).type(torch.cuda.FloatTensor)
diffusion = Diffusion(img_size=size, device=device_val)
t = diffusion.sample_timesteps(128).to(device_val)
model = UNet().to(device_val)
xmodel=model(x_input,t)
In addition to that can you please explain a bit more how is this time embedding working especially in terms of the dimension?
getting this error RuntimeError: mat1 and mat2 shapes cannot be multiplied (128x128 and 256x128)
when I am using the code below which is basically an array of size 32x32x32 and timesteps of 128:
size = 32
device_val="cuda"
x_input = torch.tensor(np.random.rand(1,size, size, size)).to(device_val).type(torch.cuda.FloatTensor)
diffusion = Diffusion(img_size=size, device=device_val)
t = diffusion.sample_timesteps(128).to(device_val)
model = UNet().to(device_val)
xmodel=model(x_input,t)
In addition to that can you please explain a bit more how is this time embedding working especially in terms of the dimension?