search

PLN-team / PLNmodels

A collection of Poisson lognormal models for multivariate count data analysis
https://pln-team.github.io/PLNmodels
GNU General Public License v3.0
54 stars 18 forks source link

More accurate message when torch is diverging. #128

Closed Bastien-mva closed 3 weeks ago

mahendra-mariadassou commented 5 months ago

Hi @Bastien-mva, the checks are not valid due to a parsing error when reading the error message. It probably has to do with nested quotes (consider using " for outer quotes and ' for inner quotes or vice-versa). Could you check and push a fix ?

Bastien-mva commented 5 months ago

Moreover, I think 0.1 is quite high for a learning rate. The learning rate has little significance with Rprop but if it is too high at the beginning it can give an infinite loss, maybe lowering it to 0.01 would be good. I can send a PR if you wish.