2020年4月7日
pytorch保存模型和导入模型以及预训练模型
摘要： "参考" model.state_dict()中保存了{参数名:参数值}的字典 保存模型 导入模型 阅读全文
posted @ 2020-04-07 23:06 happy_fan 阅读(2395) 评论(0) 推荐(0) 编辑
pytorch自定义dataset
摘要： "参考" 一个例子 输出 tensor([[ 1.3907, 0.0916], [ 0.4626, 1.3323], [ 1.4242, 2.1718], [ 1.5850, 0.3320], [ 1.0804, 0.3884], [ 0.6567, 0.1234], [ 1.6721, 0.732 阅读全文
posted @ 2020-04-07 23:02 happy_fan 阅读(1684) 评论(0) 推荐(0) 编辑