error happend when training teacher_small.pth by my self.

nelson1425 / EfficientAD

Unofficial implementation of EfficientAD https://arxiv.org/abs/2303.14535

https://arxiv.org/abs/2303.14535

Apache License 2.0

240 stars 62 forks source link

error happend when training teacher_small.pth by my self. #18

Closed shellxiao closed 10 months ago

shellxiao commented 11 months ago

it said: [RuntimeError: The size of tensor a (64) must match the size of tensor b (128) at non-singleton dimension 3 at line loss = torch.mean((target - prediction)**2)], so @nelson1425 could please tell me how to solve it? Many thanks!