Pytorch Training code flow from github

1

Hi how to understand any code obtained from github like yolov2, maskrcnn training and inference pipeline by flow diagram ? any tool available to view the flow

2

Are you looking for a tool to visualize the computation graph (model architecture)?
If so, have a look at this post, where some tools were suggested.

If not, could you explain what you mean by flow a bit more, please? :slight_smile:

3

VISUALIZING MODELS, DATA, AND TRAINING WITH TENSORBOARD

4

@ptrblck i am able to visualize the made by the link @Eta_C, i wanted to understand the training pipeline code which is either written by some one or which is available in github. Eg dataloading , preprocessing and training modules which functions are getting called and all