hello,
I am getting this error when I try this code:
source = source - torch.mean(source, dim=1, keepdim=True)
		template = template - torch.mean(template, dim=1, keepdim=True)

		output = model(template, source)
		loss_val = ChamferDistanceLoss()(template, output['transformed_source'])
		print(loss_val)
                #model.named_parameters()
		# forward + backward + optimize
		optimizer.zero_grad()
		loss_val.backward()
		optimizer.step()

print(loss_val)==>tensor(0.2791, device=‘cuda:0’, grad_fn=DivBackward0)
How can I solve this problem?
Thanks in advance,