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

optimized pytorch geometric pipeline using DDP #118

Merged
merged 495 commits into from
Jul 1, 2022
Merged

Conversation

farakiko
Copy link
Collaborator

  • fused delphes and cms pipelines through a flag --data that takes 'delphes' or 'cms'
  • optimized the torch-geometric training using DistributedDataParallel (DDP)
  • major cleanup
  • updated the cms plotting scripts to use the same plots from the tf setup

@jpata jpata merged commit 950ce04 into jpata:master Jul 1, 2022
jpata pushed a commit that referenced this pull request Sep 15, 2023
* update the pytorch pipeline

Former-commit-id: 950ce04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants