The size of tensor a (16) must match the size of tensor b (4) at non-singleton dimension 0

[04lm40]

Hello, I met a problem while running，“The size of tensor a (16) must match the size of tensor b (4) at non-singleton dimension 0.” How can I solve it? thank you.

[dongheehand]

Please let me know the command you used. It will be helpful!

[04lm40]

When I run the "### Color feature : color distribution For pokemon
python main.py --color_info dist --color_feat_dim 313 --mem_size 982 --top_k 256 --train_data_path ./pokemon --test_data_path ./test_pokemon --alpha 0.3 --color_thres 0.5 --epoch 100
"
Error is "Traceback (most recent call last):
File "/MemoPainter-PyTorch-master/main.py", line 45, in
train(args)
File "/MemoPainter-PyTorch-master/mode.py", line 73, in train
loss = mem.unsupervised_loss(res_feature, color_feat, args.color_thres)
File "/MemoPainter-PyTorch-master/memory_network.py", line 66, in unsupervised_loss
color_similarity = self.KL_divergence(color_value_expand, color_feat_expand, 1)
File "/MemoPainter-PyTorch-master/memory_network.py", line 140, in KL_divergence
log_val = torch.log10(torch.div(a, b))
RuntimeError: The size of tensor a (16) must match the size of tensor b (4) at non-singleton dimension 0"
Thank you for your answer.