[Fix] TwoTower retrain

[massquantity]

Fix #491

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.32%. Comparing base (e0da259) to head (ad80248).

Additional details and impacted files

@@           Coverage Diff           @@
##           master     #494   +/-   ##
=======================================
  Coverage   99.32%   99.32%           
=======================================
  Files         129      129           
  Lines        8448     8452    +4     
=======================================
+ Hits         8391     8395    +4     
  Misses         57       57           

Flag	Coverage Δ
CI	99.32% <100.00%> (+<0.01%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.