pytorch
c36b31d5 - `torch::nn::AdaptiveLogSoftmaxWithLoss`: check length of `cutoffs` (#106777)

Commit
1 year ago
`torch::nn::AdaptiveLogSoftmaxWithLoss`: check length of `cutoffs` (#106777) Fixes #106698 Also added a check for python API, because current error message ``` Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/home/sehoon/pytorch-latest/torch/nn/modules/adaptive.py", line 128, in __init__ or (min(cutoffs) <= 0) \ ValueError: min() arg is an empty sequence ``` is not very comprehensible. Pull Request resolved: https://github.com/pytorch/pytorch/pull/106777 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading