摘要: 译者:GeneZC torch.nn.init.calculate_gain(nonlinearity, param=None) 返回给定非线性函数的推荐的增益值。对应关系如下表: 非线性函数增益Linear / IdentityConv{1,2,3}DSigmoidTanhReLULeaky Re 阅读全文
posted @ 2019-02-17 18:47 绝不原创的飞龙 阅读(8) 评论(0) 推荐(0) 编辑
摘要: 译者:yuange250 最佳方案 保存模型的推荐方法 Pytorch主要有两种方法可用于序列化和保存一个模型。 第一种只存取模型的参数(更为推荐): 保存参数: torch.save(the_model.state_dict(), PATH) 读取参数: the_model = TheModelC 阅读全文
posted @ 2019-02-17 18:44 绝不原创的飞龙 阅读(11) 评论(0) 推荐(0) 编辑