I assume the conved.squeeze() might remove unwanted dimensions during inference, since coved has only two dimensions after the squeeze.
I’m not sure, which dimensions you would like to squeeze in this line of code, but note that all dimensions with size==1 will be removed.
If you are using a batch size of 1 during inference, this dimension will be removed, too.
I would recommend to pass the dim argument to tensor.squeeze to make sure only the desired dims are removed.