Przeglądaj źródła

Update pytorch.js

Lutz Roeder 4 lat temu
rodzic
commit
72e1c98496
1 zmienionych plików z 1 dodań i 0 usunięć
  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 {});