TylerYep / torchinfo

View model summaries in PyTorch!
MIT License
2.48k stars 117 forks source link

No verbose option #41

Closed InCogNiTo124 closed 3 years ago

InCogNiTo124 commented 3 years ago

summary(model, input_size, verbose=2, device='cpu') returns

TypeError: summary() got an unexpected keyword argument 'verbose'

InCogNiTo124 commented 3 years ago

Solved by installing from source:

git clone https://github.com/TylerYep/torchinfo.git
cd torchinfo
python3 setup.py build
sudo -E python3 setup.py install

Interestingly enough, running pip install --upgrade git+https://github.com/TylerYep/torchinfo.git somehow installed an older version. Wonder what could've went wrong

TylerYep commented 3 years ago

Hmm I've never seen this issue before. You might have installed the wrong project by accident? If I recall correctly verbose has always been available in torchinfo...