Update python.js

lutzroeder · Mar 20, 2024 · fa7f8e5 · fa7f8e5
1 parent 44d0d48
commit fa7f8e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/python.js b/source/python.js
@@ -4549,6 +4549,7 @@ python.Execution = class {
         this.registerType('torchvision.models.detection.rpn.RegionProposalNetwork', class {});
         this.registerType('torchvision.models.detection.rpn.RPNHead', class {});
         this.registerType('torchvision.models.detection.ssd.SSD', class {});
+        this.registerType('torchvision.models.detection.ssd.SSDClassificationHead', class {});
         this.registerType('torchvision.models.detection.ssd.SSDFeatureExtractorVGG', class {});
         this.registerType('torchvision.models.detection.ssdlite.SSDLiteClassificationHead', class {});
         this.registerType('torchvision.models.detection.ssdlite.SSDLiteFeatureExtractorMobileNet', class {});