Skip to content

Handle soundnet5 case where batchnorm layer is cudnn.SpatialBatchNormalization#2

Open
mohamed-ezz wants to merge 1 commit intoeborboihuc:masterfrom
mohamed-ezz:fix_loading_batchnorm_soundnet5
Open

Handle soundnet5 case where batchnorm layer is cudnn.SpatialBatchNormalization#2
mohamed-ezz wants to merge 1 commit intoeborboihuc:masterfrom
mohamed-ezz:fix_loading_batchnorm_soundnet5

Conversation

@mohamed-ezz
Copy link
Copy Markdown

The published soundnet5.t7 has cudnn.SpatialBatchNormalization layer, not nn.SpatialBatchNormalization. Add a condition to handle this.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant