search

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras
MIT License
3.98k stars 412 forks source link

pytoch-lightning not support #169

Open iambyd opened 3 years ago

iambyd commented 3 years ago

I use pytoch-lightning bulid a sample model, but get a error NotImplementedError, whether this method can support pytorch-lightning model?