Open AbrahamMulat opened 1 year ago
how to use visualize custom vision models like unet using torchview, when I try to visualize my model, it throws the following error: . . . . . RuntimeError: Failed to run torchgraph see error message
Did you find a solution ?
how to use visualize custom vision models like unet using torchview, when I try to visualize my model, it throws the following error: . . . . . RuntimeError: Failed to run torchgraph see error message