arakhmati/torchtrail

torchtrail: trace the graph of torch functions and modules for visualization, reports, etc

PythonMIT

Issues

Store arguments passed in to the TorchFunction and TorchModule and display them when passing in verbose=True to visualize
#8 opened 9 months ago by arakhmati
1
Fix the bug with`torch.cat`
#10 opened 9 months ago by arakhmati
1