Sambit Mohapatra
Results
2
issues of
Sambit Mohapatra
Hi, The training worked well. But, when I run infer.py, I get the following errors: RuntimeError: Error(s) in loading state_dict for OftNet: Missing key(s) in state_dict: "mean", "std", "frontend.conv1.weight", "frontend.bn1.weight",...
pull_loss, push_loss = _ae_loss(embd_tl, embd_br, batch['ind_masks']) File "/home/sambit/PROGRAMMING/pytorch_simple_CornerNet-master/utils/losses.py", line 84, in _ae_loss dist = F.relu(1 - (embd_mean[:, None, :] - embd_mean[:, :, None]).abs(), inplace=True) IndexError: too many indices for tensor...