returnn - 你如何提取模型中使用的初始权重？

Question

从文档中我发现，在构建模型后，权重是通过调用来初始化的TFNetwork.initialize_params。我想知道是否有 API 可以提取这些权重？

标签： returnn

Answer 1

有TFNetwork.save_params_to_file。那是你要的吗？

否则，您可以简单地单独访问所有参数。例如通过TFNetwork.get_trainable_params、或TFNetwork.get_param_values_dict、或TFNetwork.get_layer(layer_name).params等。