how to get summary of a custom model

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

3.98k stars 412 forks source link

how to get summary of a custom model #194

Open bekhzod-olimov opened 1 year ago

bekhzod-olimov commented 1 year ago

I have a custom model and I wanted to see it's summary but it results in an attribute error: AttributeError: 'CUTModel' object has no attribute 'apply'.

Model is from here.

This is the full error message: AttributeError: 'CUTModel' object has no attribute 'apply'.

How to resolve it?