search

snuspl / parallax

A Tool for Automatic Parallelization of Deep Learning Training in Distributed Multi-GPU Environments.
Apache License 2.0
130 stars 35 forks source link

Add Profiling Option #7

Closed sj6077 closed 6 years ago

sj6077 commented 6 years ago

Things to Change

Add Profiling option into ParallaxConfig.

Current Behavior

Expected Behavior

Save RunMetadata to the profile_dir when the global_step is one of profile_steps

Failure Information (for bugs)

Failure Logs

How to Reproduce

Related Issues