Skip to content

Adding ParallelBlock#1088

Merged
marcromeyn merged 10 commits intomainfrom
torch/parallel-block
May 22, 2023
Merged

Adding ParallelBlock#1088
marcromeyn merged 10 commits intomainfrom
torch/parallel-block

Conversation

@marcromeyn
Copy link
Contributor

@marcromeyn marcromeyn commented May 12, 2023

Goals ⚽

Adding ParallelBlock.

Implementation Details 🚧

The definition of this block is slightly different than we are used to in TF. In this implementation each branch is of type BlockContainer, this makes it easier to append/prepend to it.

Testing Details 🔍

Since it's a low-level class, I maximized the test-coverage
CleanShot 2023-05-12 at 15 59 57@2x

@github-actions
Copy link

Documentation preview

https://nvidia-merlin.github.io/models/review/pr-1088

@marcromeyn marcromeyn force-pushed the torch/parallel-block branch from 77ca69b to 8ce54d1 Compare May 17, 2023 07:35
@marcromeyn marcromeyn marked this pull request as ready for review May 17, 2023 07:39
@marcromeyn marcromeyn requested a review from edknv May 17, 2023 07:39
@marcromeyn marcromeyn added this to the Merlin 23.06 milestone May 17, 2023
@marcromeyn marcromeyn merged commit 67719df into main May 22, 2023
@marcromeyn marcromeyn deleted the torch/parallel-block branch May 22, 2023 15:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants