Ver Fonte

Update pytorch.js

Lutz Roeder há 6 anos atrás
pai
commit
285b16493e
1 ficheiros alterados com 2 adições e 0 exclusões
  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() {});