slot-attention-pytorch icon indicating copy to clipboard operation
slot-attention-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon evelinehong

Added reparam trick and log-sigma

Open shahrushi2003 opened this issue 1 year ago • 0 comments

Resolved the non-trainability of mu and sigma through the reparameterization trick. Also replaced sigma with log(sigma) for numerical stability.

shahrushi2003 avatar Oct 05 '23 11:10 shahrushi2003