Question

我正在尝试恢复这个NN的参数：

    nn.Sequential {
      [input -> (1) -> (2) -> (3) -> (4) -> (5) -> (6) -> output]
      (1): nn.Linear(4 -> 200)
      (2): nn.Tanh
      (3): nn.Linear(200 -> 200)
      (4): nn.Tanh
      (5): nn.Linear(200 -> 3)
      (6): nn.LogSoftMax         
    }

使用此代码：

print(mlp:get(1).weight)
print(mlp:get(1).bias)

print(mlp:get(3).weight)
print(mlp:get(3).bias)

print(mlp:get(5).weight)
print(mlp:get(5).bias)

使用以下命令行将输出.lua文件保存到文本文件中：

>>th  'MyScript.lua' > NNParameters.txt

我将所有重量矩阵包裹成每列六列（第1列到第6列......第193到198列......第199到第200列）。

无论如何都要阻止文本被包装并仅在一个块中显示权重矩阵？

谢谢。

Answer 1

Make generated parameters final

Answer 2

我认为你真正想要做的是保存参数，以便以后加载它们？在这种情况下，请看这个：

https://github.com/torch/torch7/blob/master/doc/serialization.md

torch7：文本文件中的打印矩阵，行长度超过80个字符

2 个答案: