Fixed DLRModel / DLRMBlock to accept bottom_block with dropout on top of MLPs#1018

Merged

gabrielspmoreira merged 2 commits intomainfrom tf/dlrm_dropout_fix

Mar 15, 2023

+13-10

Commits on Mar 14, 2023

Fixed DLRMBlock to accept bottom_block with dropout or other layers on top of MLPs
gabrielspmoreira
committed

Commits on Mar 15, 2023

Merge branch 'main' into tf/dlrm_dropout_fix
gabrielspmoreira
authored