pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

[BUG] Bug in NextVit reparameterize

Open chuong98 opened this issue 1 year ago • 1 comments

Describe the bug When I do reparameterize the model NextViT for onnx export, it returns the error: self.norm(x). None Object is not callable.

I believe line 200 of file https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/nextvit.py should be self.norm = nn.Identity() instead of self.norm = None

chuong98 avatar May 24 '24 19:05 chuong98

@chuong98 yeah, that's wrong, but also there is an issue wrt to checking for None vs isinstance(nn.Identity()) .. need look through it more closely

rwightman avatar May 24 '24 23:05 rwightman