weights.py 文件源码

python
阅读 33 收藏 0 点赞 0 评论 0

项目:paysage 作者: drckf 项目源码 文件源码
def load_params(self, store, key):
        """
        Load the parameters from an HDFStore.

        Notes:
            Performs an IO operation.

        Args:
            store (pandas.HDFStore): the readable stream for the params.
            key (str): the path for the layer params.

        Returns:
            None

        """
        params = []
        for i, ip in enumerate(self.params):
            params.append(be.float_tensor(
                store.get(os.path.join(key, 'parameters', 'key'+str(i))).as_matrix()
            ).squeeze()) # collapse trivial dimensions to a vector
        self.params = self.params.__class__(*params)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号