OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix(module): remove assert during forward 

GitOrigin-RevId: bb04d861a3
release-1.5
Megvii Engine Team huangxinda 3 years ago
parent
427113088b
commit
0c97b2a3ec
1 changed files with 0 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +0
    -1
      imperative/python/megengine/module/normalization.py

+ 0
- 1
imperative/python/megengine/module/normalization.py View File

@@ -136,7 +136,6 @@ class LayerNorm(Module):


def forward(self, x): def forward(self, x):
x_shape = x.shape x_shape = x.shape
assert x_shape[-len(self.normalized_shape) :] == self.normalized_shape
dim_delta = len(x_shape) - len(self.normalized_shape) dim_delta = len(x_shape) - len(self.normalized_shape)
non_flatten_shape = x_shape[:dim_delta] non_flatten_shape = x_shape[:dim_delta]
x = x.reshape(*non_flatten_shape, -1) x = x.reshape(*non_flatten_shape, -1)


Write Preview
Loading…
Cancel
Save