Quellcode durchsuchen

Update pytorch.js

Lutz Roeder vor 6 Jahren
Ursprung
Commit
285b16493e
1 geänderte Dateien mit 2 neuen und 0 gelöschten Zeilen
  1. 2 0
      src/pytorch.js

+ 2 - 0
src/pytorch.js

@@ -1142,6 +1142,8 @@ pytorch.Execution = class {
         this._registerConstructor('torchvision.models.mobilenet.InvertedResidual', function() {});
         this._registerConstructor('torchvision.models.resnet.Bottleneck', function() {});
         this._registerConstructor('torchvision.models.resnet.BasicBlock', function() {});
+        this._registerConstructor('torchvision.models.quantization.resnet.QuantizableBottleneck', function() {});
+        this._registerConstructor('torchvision.models.quantization.resnet.QuantizableResNet', function() {});
         this._registerConstructor('torchvision.models.segmentation.deeplabv3.ASPP', function() {});
         this._registerConstructor('torchvision.models.segmentation.deeplabv3.ASPPConv', function() {}); 
         this._registerConstructor('torchvision.models.segmentation.deeplabv3.ASPPPooling', function() {});