Lutz Roeder 5 лет назад
Родитель
Сommit
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); };