瀏覽代碼

Update torch.js

Lutz Roeder 5 年之前
父節點
當前提交
a95c23ba78
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      source/torch.js

+ 1 - 0
source/torch.js

@@ -773,6 +773,7 @@ torch.T7Reader = class {
         this._registry['nn.SpatialFullConvolution'] = function(reader) { reader.nn(this); };
         this._registry['nn.SpatialLPPooling'] = function(reader) { reader.nn(this); };
         this._registry['nn.SpatialMaxPooling'] = function(reader) { reader.nn(this); };
+        this._registry['nn.SpatialMaxUnpooling'] = function(reader) { reader.nn(this); };
         this._registry['nn.SpatialReflectionPadding'] = function(reader) { reader.nn(this); };
         this._registry['nn.SpatialReplicationPadding'] = function(reader) { reader.nn(this); };
         this._registry['nn.SpatialSoftMax'] = function(reader) { reader.nn(this); };