将火炬t7型号转换为keras h5

作者: 哦豁
发布时间: 2024-05-14 03:10:47 (2月前)
转自：

2 条回复

0#
回复此人
757461156 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 如此链接所述， <a href="https://github.com/pytorch/pytorch/issues/15307#issuecomment-448086741" rel="nofollow noreferrer"> https://github.com/pytorch/pytorch/issues/15307#issuecomment-448086741 </A> </p> <P> 同 <EM> torchfile </EM> 包，负载成功。你可以转储的内容 <EM> 模型 </EM> 到一个文件，然后了解内容。每个层信息都存储为字典。了解模型体系结构可以更容易地解析内容。 </p> <pre> <code> >>> import torchfile >>> model = torchfile.load('c3d-sports1m-kinetics.t7') >>> module = model.modules[0].modules[0] >>> module.name b'conv1a' >>> module['weight'].shape (64, 3, 3, 3, 3) >>> module['bias'].shape (64,) </code> </pre> </DIV>

编辑

登录后才能参与评论