search

sksq96 / pytorch-summary

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

summary() got an unexpected keyword argument 'dtypes' #193

Open satabios opened 1 year ago

satabios commented 1 year ago

summary(model.to(device),(3,640,640), dtypes= [torch.half])

returns

summary() got an unexpected keyword argument 'dtypes'