Lutz Roeder преди 4 години
родител
ревизия
72e1c98496
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      source/pytorch.js

+ 1 - 0
source/pytorch.js

@@ -999,6 +999,7 @@ pytorch.Execution = class extends python.Execution {
         this.registerType('torch.nn.modules.batchnorm.BatchNorm1d', class {});
         this.registerType('torch.nn.modules.batchnorm.BatchNorm2d', class {});
         this.registerType('torch.nn.modules.batchnorm.BatchNorm3d', class {});
+        this.registerType('torch.nn.modules.batchnorm.LazyBatchNorm1d', class {});
         this.registerType('torch.nn.modules.batchnorm.SyncBatchNorm', class {});
         this.registerType('torch.nn.modules.container.ModuleDict', class {});
         this.registerType('torch.nn.modules.container.ModuleList', class {});