nkh

2017年10月12日

pytorch加载和保存模型

摘要: 在模型完成训练后,我们需要将训练好的模型保存为一个文件供测试使用,或者因为一些原因我们需要继续之前的状态训练之前保存的模型,那么如何在PyTorch中保存和恢复模型呢? 方法一(推荐): 第一种方法也是官方推荐的方法,只保存和恢复模型中的参数。 保存 torch.save(the_model.sta 阅读全文

posted @ 2017-10-12 16:16 nkh 阅读(32181) 评论(0) 推荐(0) 编辑

python中的__call__()函数

摘要: __call__ 在Python中,函数其实是一个对象: >>> f = abs >>> f.__name__ 'abs' >>> f(-123) 123 由于 f 可以被调用,所以,f 被称为可调用对象。 所有的函数都是可调用对象。 一个类实例也可以变成一个可调用对象,只需要实现一个特殊方法__c 阅读全文

posted @ 2017-10-12 11:26 nkh 阅读(998) 评论(0) 推荐(0) 编辑

导航