Forráskód Böngészése

Update pytorch.js

Lutz Roeder 6 éve
szülő
commit
4c2a0daf51
1 módosított fájl, 6 hozzáadás és 0 törlés
  1. 6 0
      src/pytorch.js

+ 6 - 0
src/pytorch.js

@@ -1124,6 +1124,9 @@ pytorch.Execution = class {
         this._registerConstructor('torchvision.models.detection.faster_rcnn.FasterRCNN', function() {});
         this._registerConstructor('torchvision.models.detection.faster_rcnn.FastRCNNPredictor', function() {});
         this._registerConstructor('torchvision.models.detection.faster_rcnn.TwoMLPHead', function() {});
+        this._registerConstructor('torchvision.models.detection.mask_rcnn.MaskRCNN', function() {});
+        this._registerConstructor('torchvision.models.detection.mask_rcnn.MaskRCNNHeads', function() {});
+        this._registerConstructor('torchvision.models.detection.mask_rcnn.MaskRCNNPredictor', function() {});
         this._registerConstructor('torchvision.models.detection.roi_heads.RoIHeads', function() {});
         this._registerConstructor('torchvision.models.detection.rpn.AnchorGenerator', function() {});
         this._registerConstructor('torchvision.models.detection.rpn.RegionProposalNetwork', function() {});
@@ -1339,6 +1342,9 @@ pytorch.Execution = class {
             }
             throw new pytorch.Error("Unsupported bytearray encoding '" + JSON.stringify(encoding) + "'.");
         });
+        this._registerFunction('__builtin__.slice', function(start, stop , step) {
+            return [ start, stop, step ];
+        });
         this._registerFunction('collections.Counter', function(/* iterable */) {
             return {};
         });