pyro-ppl/pyro

Feature request: Prohibit passing negative numbers to plate sizes

Ouverte

#3 407 ouverte le 24 oct. 2024

 (0 commentaire) (0 réaction) (0 personne assignée)Python (981 forks)batch import
help wantedusability

Métriques du dépôt

Stars
 (8 211 étoiles)
Métriques de merge PR
 (Merge moyen 10j 19h) (1 PR mergée en 30 j)

Description

Issue Description

Plate primitives scale log-likelihood terms by size/batch_size. But plates allow negative numbers without a warning or assert. Something like assert size > 0 above https://github.com/pyro-ppl/pyro/blob/dev/pyro/primitives.py#L424 might be worth adding.

Guide contributeur