Closed maltelueken closed 1 week ago
When using a continuous approximator with flow matching plus set transformer (or deepset) and base_distribution="normal", the loss becomes nan after two steps.
base_distribution="normal"
nan
Is there anything that I missed to add MMD to the summary loss?
Thanks for reporting the issue. Could you please provide a minimal reproducible example, at least of the exact definitions of your networks?
Closing this as not reproducible.
When using a continuous approximator with flow matching plus set transformer (or deepset) and
base_distribution="normal"
, the loss becomesnan
after two steps.Is there anything that I missed to add MMD to the summary loss?