Update pytorch.js

lutzroeder · Dec 9, 2021 · 4aede5f · 4aede5f
1 parent d066d9d
commit 4aede5f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/pytorch.js b/source/pytorch.js
@@ -1233,6 +1233,7 @@ pytorch.Execution = class extends python.Execution {
         this.registerType('torchvision.ops.feature_pyramid_network.FeaturePyramidNetwork', class {});
         this.registerType('torchvision.ops.feature_pyramid_network.LastLevelMaxPool', class {});
         this.registerType('torchvision.ops.feature_pyramid_network.LastLevelP6P7', class {});
+        this.registerType('torchvision.ops.misc.ConvNormActivation', class {});
         this.registerType('torchvision.ops.misc.ConvTranspose2d', class {});
         this.registerType('torchvision.ops.misc.FrozenBatchNorm2d', class {});
         this.registerType('torchvision.ops.poolers.LevelMapper', class {});