We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
When instantiating the Cortex class, the function apply_circular_mask_to_weights() does:
Cortex
apply_circular_mask_to_weights()
tensor = circular_mask(matrix.size()[1], matrix.size()[0], radius, matrix.is_cuda) matrix.masked_fill_(tensor, 0)
However, in PyTorch 1.3.1 (and likely some previous versions), calling masked_fill_ with a tensor that isn't a BoolTensor is depreciated.
masked_fill_
BoolTensor
The warning can be eliminated by changing the line to:
matrix.masked_fill_(tensor.type(torch.BoolTensor), 0)
though a better solution would be to have all the mask functions (like circular_mask()) return BoolTensor instead.
circular_mask()
The text was updated successfully, but these errors were encountered:
No branches or pull requests
When instantiating the
Cortexclass, the functionapply_circular_mask_to_weights()does:However, in PyTorch 1.3.1 (and likely some previous versions), calling
masked_fill_with a tensor that isn't aBoolTensoris depreciated.The warning can be eliminated by changing the line to:
though a better solution would be to have all the mask functions (like
circular_mask()) returnBoolTensorinstead.The text was updated successfully, but these errors were encountered: