摘要: 今天重新研究了一下pytorch如何自定义权重,可以根据条件筛选赋值,也可以根据自定义某个张量赋值,还可以根据pth文件部分赋值等情况, 我总结了三种方法(self.modules(),self.state_dict(),self.named_parameters()),每种方法有约2种情况,希望对大家有所帮助, 然具体各自函数代表什么,可自行查阅,如self.parameters函数,我给出具体实现权重的初始化方法(已使用代码验证)。 阅读全文
posted @ 2020-09-25 17:05 tangjunjun 阅读(10332) 评论(0) 推荐(1) 编辑
https://rpc.cnblogs.com/metaweblog/tangjunjun