search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

Do not support pytorch>=0.4.1? #25

Closed wangg12 closed 5 years ago

wangg12 commented 5 years ago

@szagoruyko This will be failed: assert all(isinstance(p, Variable) for p in params.values())

Treyoo commented 5 years ago

Obviously yes

szagoruyko commented 5 years ago

Tensor is Variable, it will not faill