add profiling support to sg_train

buriburisuri / sugartensor

A slim tensorflow wrapper that provides syntactic sugar for tensor variables. This library will be helpful for practical deep learning researchers not beginners.

MIT License

372 stars 63 forks source link

add profiling support to sg_train #30

Open AndreasMadsen opened 7 years ago

AndreasMadsen commented 7 years ago

By setting profile = global_step in sg_train a timeline tracing dump will be added to the save directory.