issue when computing log-cost for data -> data bridge training

cesarali / conditional_rate_matching

Flow Matching for Discrete Variables

2 stars 0 forks source link

issue when computing log-cost for data -> data bridge training #5

Closed dfaroughy closed 3 months ago

dfaroughy commented 3 months ago

dimensions not matching when repeating tensor during computation of the outer prob in the log-cost function. It only happens when bridging two datasets. It works fine when starting from noise -> data.

https://github.com/cesarali/conditional_rate_matching/blob/0927ec20bc054e26cd05e9d0b346a8646a04d8b7/src/conditional_rate_matching/models/temporal_networks/rates/crm_rates.py#L286

cesarali commented 3 months ago

Done