An official implementation of "Learning Memory-guided Normality for Anomaly Detection" (CVPR 2020) in PyTorch.
333
stars
82
forks
source link
RuntimeError: The size of tensor a (3) must match the size of tensor b (516) at non-singleton dimension 0 #33
Open
efsunsezer opened 3 years ago
Hi when I use ped2 (args.method == 'pred) in Evaluate.py following line
torch.mean(loss_func_mse((outputs[0]+1)/2, (imgs[0,3*4:]+1)/2)).item() causes
RuntimeError: The size of tensor a (3) must match the size of tensor b (516) at non-singleton dimension 0.