Skip to content

A pytorch version work achieving PFN model for Particle Physics Using DistributedDataParallel method

Notifications You must be signed in to change notification settings

Wuyoufei/ParticleFlowNetwork_DDP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

###A pytorch script to achieve ParticleFlowNetwork using DistributedDataParallel method

script_GPUonly and script_GPUorCPU are same but the latter will slower than GPUonly, so Please use script_GPUonly when sbatch the script.

to run the script: just ./start_GPU.sh check the settings and type a suffix for this training then l means running locally(for test), s means sbatch

Please change the hyperparameters in the

script_GPUonly/my_train_DDP.py

About

A pytorch version work achieving PFN model for Particle Physics Using DistributedDataParallel method

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published