Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding ParallelBlock #1088

Merged
merged 10 commits into from
May 22, 2023
Merged

Adding ParallelBlock #1088

merged 10 commits into from
May 22, 2023

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
.github/workflows/pytorch.yml Outdated Show resolved Hide resolved
merlin/models/torch/block.py Outdated Show resolved Hide resolved
merlin/models/torch/container.py Outdated Show resolved Hide resolved
@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