You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
If I want to call draw_graph(model, input_size=..., depth=1) what should the input_size look like? Is it supported?
I believe the draw_graph function could handle a input_size in a list meaning the forward will receive as many arguments as element in the list passed.
The text was updated successfully, but these errors were encountered:
Yes it is supported. input_size is either SizeItem or list(SizeItem) where SizeItem is anything that can represent the shape of a torch Tensor, e.g. tuple, torch.Size.
Imagine I have a module like this:
If I want to call
draw_graph(model, input_size=..., depth=1)
what should the input_size look like? Is it supported?I believe the draw_graph function could handle a input_size in a list meaning the forward will receive as many arguments as element in the list passed.
The text was updated successfully, but these errors were encountered: