pytorch/torch/_refs
Gonçalo Rua 139647d317 Fix #83241: torch.nn.TripletMarginLoss allowed margin less or equal to 0 (#121978)
Documentation states that the parameter margin of torch.nn.TripletMarginLoss is greater than 0, however any value was being accepted. Also fixed torch.nn.TripletMarginWithDistanceLoss which had the same problem. Added error test input for the new ValueError.

Fixes #83241

Pull Request resolved: https://github.com/pytorch/pytorch/pull/121978
Approved by: https://github.com/mikaylagawarecki
2024-03-19 23:19:11 +00:00
..
linalg Fix Dynamo tests failing with "Failed running call_function <built-in function linalg_norm" (#120993) 2024-03-01 20:27:45 +00:00
nn Fix #83241: torch.nn.TripletMarginLoss allowed margin less or equal to 0 (#121978) 2024-03-19 23:19:11 +00:00
special
__init__.py better support set_default_dtype(torch.float16), update doc (#121730) 2024-03-15 06:48:42 +00:00
_conversions.py
fft.py